From a list
的值,我尝试识别总和超过 10 的任何连续值对
a = [1,9,3,4,5]
...所以我写了一个for
loop...
values = []
for i in range(len(a)-2):
if sum(a[i:i+2]) >10:
values += [a[i:i+2]]
...我将其重写为列表理解...
values = [a[i:i+2] for i in range(len(a)-2) if sum(a[i:i+2]) >10]
两者产生相同的输出:
values = [[1,9], [9,3]]
我的问题是如何最好地在 DataFrame 中应用上述列表理解。
这是 5 行 DataFrame 示例
import pandas as pd
df = pd.DataFrame({'A': [1,1,1,1,0],
'B': [9,8,3,2,2],
'C': [3,3,3,10,3],
'E': [4,4,4,4,4],
'F': [5,5,5,5,5]})
df['X'] = df.values.tolist()
在哪里:
- a 在 a 之内df['X']
这是一个值列表 A - F 列
df['X'] = [[1,9,3,4,5],[1,8,3,4,5],[1,3,3,4,5],[1,2,10,4,5],[0,2,3,4,5]]
期望的输出是:
df['X1'] = [[[1,9], [9,3]],[[8,3]],[[NaN]],[[2,10],[10,4]],[[NaN]]]
谢谢。