python - 选择列表的各种 "Pieces"

Question

我在 Pandas DataFrame 中有一个列列表，我希望在没有手动输入的情况下创建某些列的列表。

我的问题是我正在学习，但知识还不够。

我曾尝试在互联网上搜索，但没有找到适合我的情况。如果有重复，我深表歉意。

我试图从中删除的列表如下所示:

['模型',
'显示',
'圆柱体'，
'反式',
'驾驶'，
'燃料'，
'veh_class',
'air_pollution_score',
'city_mpg',
'hwy_mpg',
'cmb_mpg',
'greenhouse_gas_score',
'smartway']

这是我自己写的代码:dataframe.columns.tolist()[:6,8:10,11]

在这种情况下，我试图选择除“air_pollution_score”和“greenhouse_gas_score”之外的所有内容

我的最终目标是了解语法以及如何选择列表的各个部分。

Answer 1

您可以这样做，或者您可以只使用 drop 删除您不想要的列:

dataframe.drop(['air_pollution_score', 'greenhouse_gas_score'], axis=1).columns

请注意，您需要指定 axis=1 以便 pandas 知道您要删除列，而不是行。

即使您想使用list 语法，我还是会说最好使用list 理解；像这样:

exclude_columns = ['air_pollution_score', 'greenhouse_gas_score']

[col for col in dataframe.columns if col not in exclude_columns]

这会获取数据框中的所有列除非它们出现在exclude_columns 中。