python - 用 Pandas 查找两列或多列的最大值

Question

我有一个包含 A、B 列的数据框。我需要为每条记录/行创建一个列 C:

C = max(A, B).

我该怎么做呢？

Answer 1

你可以像这样得到最大值:

>>> import pandas as pd
>>> df = pd.DataFrame({"A": [1,2,3], "B": [-2, 8, 1]})
>>> df
   A  B
0  1 -2
1  2  8
2  3  1
>>> df[["A", "B"]]
   A  B
0  1 -2
1  2  8
2  3  1
>>> df[["A", "B"]].max(axis=1)
0    1
1    8
2    3

等等:

>>> df["C"] = df[["A", "B"]].max(axis=1)
>>> df
   A  B  C
0  1 -2  1
1  2  8  8
2  3  1  3

如果您知道“A”和“B”是唯一的列，您甚至可以侥幸逃脱

>>> df["C"] = df.max(axis=1)

我猜你也可以使用 .apply(max, axis=1)。