python - pandas DataFrame - 查找偏移列之间的最大值

Question

假设我有一个由以下给出的 pandas 数据框

import pandas as pd
import numpy as np

df = pd.DataFrame(np.random.randn(5,2))
df

    0           1
0   0.264053    -1.225456
1   0.805492    -1.072943
2   0.142433    -0.469905
3   0.758322    0.804881
4   -0.281493   0.602433

我想返回一个包含 4 行的 Series 对象，其中包含 max(df[0,0], df[1,1]), max(df[1,0], df[2,1]), max(df[2,0], df[3,1]), max(df[3,0], df[4,1]) 。更一般地说，比较列 0 和列 1 偏移量 n 行的最大值的最佳方法是什么？

谢谢。

Answer 1

您希望在移动第一列后将 max 应用于行。

pd.concat([df.iloc[:, 0].shift(), df.iloc[:, 1]], axis=1).apply(max, axis=1).dropna()