pandas习题 043:移动窗口中最大值对应的其他列值
编码题)有以下 DataFrame 名为 df,从 A 列的第一个数据开始,每次取三个连续的数据(不够三个按当前小于三个的数据算),找出这三个数据中的最大值,并找到这个最大值在B列中对应的值。产生新 C 列数据。
import pandas as pd
df = pd.DataFrame({
‘A’: [10, 20, 15, 30, 25, 40, 35, 50, 45],
‘B’: [1, 2, 3, 4, 5, 6, 7, 8, 9]
})
df
‘’’
A B
0 10 1
1 20 2
2 15 3
3 30 4
4 25 5
5 40 6
6 35 7
7 50 8
8 45 9
‘’’
即结果为:
‘’’
A B C
0 10 1 1
1 20 2 2
2 15 3 2
3 30 4 4
4 25 5 4
5 40 6 6
6 35 7 6
7 50 8 8
8 45 9 8
‘’’
如果用迭代输出:
for i in<