Как определить имена и количество столбцов в кадре данных на основе значения параметра функции в Python Pandas?

avatar
koler
8 августа 2021 в 16:54
48
1
0

У меня есть собственная функция сборки в Python Pandas, как показано ниже (но, очевидно, это всего лишь пример):

def my_function(x):
    df = pd.DataFrame(columns=x)

И мне нужно сделать что-то вроде этого, например:

  • если параметр x равен 4, то имена столбцов: "col1", "col2", "col3", "col4" и всего 4 столбца
  • если параметр x равен 2, то имена столбцов: "col1", "col2" и есть 2 столбца

Поэтому имя столбца (col...) и количество столбцов должны зависеть от значения параметра x этой функции.

Как я могу это сделать? Как изменить эту функцию, чтобы добиться ее в Python Pandas?

Источник
mozway
24 августа 2021 в 06:52
0

если вы не против проголосовать за мой ответ, это будет оценено по достоинству, поскольку за него проголосовали без причины

Ответы (1)

avatar
mozway
8 августа 2021 в 16:57
-1

вы можете использовать:

def my_function(x):
    df = pd.DataFrame({'col%s' % (i+1): [] for i in range(x)})
koler
8 августа 2021 в 17:04
0

Отлично работает! а что значит "s" в 'col%s...." ? Я вижу, что без этого "s" не работает ?

mozway
8 августа 2021 в 17:08
1

Это делается для того, чтобы отформатировать строку '%s', которая заменяется числом. Вы можете прочитать это: pyformat.info