截断 Pandas 数据框中固定数量的列

Question

我有一个数据框，在某些情况下可能会增长到大量列。有时，出于显示目的，列数会变得过多。我想做的是，数据框可能不超过 16 列，如果超过，则截断多余的列。

Pandas 有截断功能。默认情况下，它适用于行，但也可以适用于列。看起来就像

df.truncate(before="A", after="B", axis="columns")

假设 A 和 B 都是数据帧中的列。但是，如果我像

df.truncate(before=1, after=16, axis="columns")

那样，我会遇到切片问题，因为它显然需要列名称。

有没有办法在不知道或不关心列名称的情况下删除多余的列？这很重要，因为它是通用代码，可用于不同但相关的数据。

我尝试使用截断函数，我希望能够通过固定常量限制列数。

Answer 1

如果您的列名称是唯一的，您可以使用：

df.truncate(before=df.columns[1], after=df.columns[16], axis="columns")