набор столбец pandas.DataFrame объекта

Question

В идеале, я хочу, чтобы иметь возможность что-то вроде:

cols = ['A', 'B', 'C'] 
df = pandas.DataFrame(index=range(5), columns=cols) 
df.get_column(cols[0]) = [1, 2, 3, 4, 5] 

Что такое вещий/pandonic способ сделать это?

Редактировать: Я знаю, что могу получить доступ к столбцу «А» на df.A, но в целом я не знаю, что такое имена столбцов.

Answer 1

Вам не нужно хранить какие столбцы у DataFrame отдельно.

Вы можете узнать, какие столбцы существуют в панд DataFrame путем доступа к переменной DataFrame.columns.

Чтобы получить доступ к серии прикрепленной к определенному колонку, вы можете использовать метод GetItem в DataFrame []

Крошечного пример:

col = df.columns[0] 
df[col] = [1, 2, 3, 4, 5] 

Answer 2

Хорошо, это особенно просто.

df[cols[0]] = [1, 2, 3, 4, 5]