python - Python，Pandas：Datframe - 保留而不是删除列

Question

早上好，

我一直在使用的数据框（100%）包含四个特定的列，但其余列各不相同！有没有办法只保留某些列而不删除它们？现在我正在使用此代码执行...

df = df.drop(["column_a", "column_b", "column_c", axis=1)

...但是每次更改列的数量时，我都必须调整我的代码。

谢谢，周末愉快！

标签： pythonpandas

Answer 1

df = df[['col_to_keep_1', 'col_to_keep_2', 'col_to_keep_3', 'col_to_keep_4']]