将列范围指定为参数

Question

我想提取具有100行的表的最大值:可以是很差的的值，如下所示：

maxA = max(data['a1size'].max(), data['a2size'].max(), data['a3size'].max()... and continuing like this until a100size.

当然，这一解决办法根本不是理想的。

有比使用for循环更好、更优雅的方法吗？

我在找某种..操作符。类似于：

maxA = max(data['a1size'].max() .. data['a100size'].max())

Answer 1

loc运算符允许您使用按列名进行切片：

maxA = data.loc[:, 'a1size':'a100size'].values.max()

请注意，与核心Python切片不同的是，这里包含了最后一列。第一个:指定需要所有行。

如果要取最大值的列是连续的，则这是可行的。否则，您可以用列名列表来索引dataframe。要构造这样的列表，可以使用列表理解：

a_size_cols = [f'a{i}size' for i in range(1, 101)]

maxA = data[a_size_cols].values.max()