如何 select 子數據幀基於多個 column_name 列表

Question

我有一個包含大量列的 dataframe。 我對這些列進行了分類，得到了三個列表，每個列表都包含一些列名。 例如：

l1 = ['Year','Day','Month']

l2 = ['A1','A2','A3',...,'A50']

l3 = ['B1','B2','B3',...,'B50']

我想要一個新的 dataframe 只包含上述三個列表中的列。

我試過了

df_new = df[l1] + df[l2] + df[l3]

但結果不正確，因為數據類型不同。 我不認為merge 、 join或concatenate是必要的，因為列的長度是相同的，並且它們沒有任何共同的列名。 你能幫我一個方便的方法嗎？

Answer 1

您需要先添加列表以獲取單個列表中的所有列名，然后使用它來獲取 dataframe 的一部分。 嘗試：

df_new = df[[l1 + l2 + l3]]