python - 让熊猫识别包含与其数据框中的列相同的列名的列表？

Question

假设我有一个名为 x 的列表

x = ['Sales', 'Total', 'Quantity']

我有一个 excel 数据框，其中包含名为“Employee”、“Age”、“Sex”、“Sales”、“Quantity”和“Total”的列。如何让熊猫只选择与列表中名称相同的数据框列？

标签： pythonpandas

Answer 1

做就是了：

x = ['Sales', 'Total', 'Quantity']

df = df[x]

由于x已经是列列表，因此在内部使用它single-brackets来对数据框进行子集化。

或使用Index.intersection：

df = df[df.columns.intersection(x)]