求教如何用简短代码发现一个 dataframe 里出现 a 列值大于相邻行的 b 列值情况的位置。。。

2020-08-19 20:14:10 +08:00
 yellowtail

https://imgchr.com/i/d1DavD

所有的 low 列的值不应该高于上下相邻行的 high 列的值 想过分别取两列值作为两个 list 然后相减但是觉得得减两次(上下关系各一次) 而且保留序号也很麻烦...

1335 次点击
所在节点    Python
5 条回复
wanv1171
2020-08-20 04:34:30 +08:00
用 shift?

df['Last_High'] = df['High'].shift(1)
df['Next_High'] = df['High'].shift(-1)

df['Low_Bigger_Than_High'] = df.apply(lambda row: row['Low'] > max(row['High'], row['Last_High'], row['Next_High']),axis = 1)
no1xsyzy
2020-08-20 09:30:30 +08:00
shift 完可以整列进行比较、布尔运算,然后根据布尔值筛行
apply 岂不是又回到 python 里运算了
princelai
2020-08-20 10:24:02 +08:00
condition = (data.low < data.high)&(data.low < data.high.shift(1))&(data.low < data.high.shift(-1))
data.loc[condition]
yellowtail
2020-08-20 16:18:46 +08:00
@princelai 对不起...昨天发帖的时候不太清醒...我的意思是所有 mark 列值为 1 的 high 列值 应大于 相邻 mark 列值为-1 的 low 列值..
yellowtail
2020-08-20 16:32:54 +08:00
@princelai 已解决,在 condition 加上[2::2]..

这是一个专为移动设备优化的页面(即为了让你能够在 Google 搜索结果里秒开这个页面),如果你希望参与 V2EX 社区的讨论,你可以继续到 V2EX 上打开本讨论主题的完整版本。

https://www.v2ex.com/t/699724

V2EX 是创意工作者们的社区,是一个分享自己正在做的有趣事物、交流想法,可以遇见新朋友甚至新机会的地方。

V2EX is a community of developers, designers and creative people.

© 2021 V2EX