Как получить атрибуты dataframe X из списка?

Question

У меня есть фрейм данных X с 58 столбцами и список некоторых имен столбцов X. Я хочу получить столбцы из X, используя мой список, такой как X.my_list [index], но я не могу, потому что my_list [index] возвращаетимя столбца, подобное следующему: 'column_name'.Однако единственный способ, которым это работает, является X.column_name.Короче говоря, мне нужно получить имена столбцов из моего списка, чтобы использовать их на моем фрейме данных X без '', например, X.column_name.Как я могу справиться с этим?

Answer 1

Вы можете использовать следующее,

X [ "column_name"]

чтобы получить столбцы в форме Series.

# Example
X = pd.DataFrame(data=[[1, "A", 1.1], [2, "B", 2.2], [3, "C", 3.3]], columns=["a", "b", "c"])
col_list=["a", "b", "c"]

for col in col_list:
   print(X[col])