如何在熊猫数据框中删除多个（〜5000）列？

Question

我有一个包含5632列的数据框，我只想保留其中的500列。我也有一个数据框中的列名（我想保留），并将这些列名作为行索引。有什么办法吗？

Answer 1

让我们假设您的DataFrame被命名为df，并且您有一个要保留的列索引列表cols。然后，您应该使用：

df1 = df.iloc[:, cols]

此语句将删除除在cols中已指定其索引的列以外的所有列。使用df1作为新的DataFrame。