微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

如何根据某些条件编辑数据框?

如何解决如何根据某些条件编辑数据框?

我绘制了两个数据框,类似于 this。我想获得一个新的数据框,这样每当这两条曲线相互交叉时,只需在这些位置放置 1 或 -1。当蓝色曲线与橙色曲线相交时,正号可能是向上和相反的 -1。

使用此代码生成的数据框(供参考):

df=pd.DataFrame()
df['curve1'] = pd.DataFrame(np.sin([a/(2*np.pi) for a in range(180)])*np.random.choice([1,.8,1.2],180,p=(.5,.25,.25)),columns=["data"])
df['curve2'] = pd.DataFrame(np.sin([-a/(2*np.pi) for a in range(180)])*np.random.choice([1,columns=["data"])

首先我正在计算这些曲线之间的差距-

df['curve_diff'] = df['curve1']-df['curve2']

接下来,我使用以下命令。

df.loc[df['curve_diff'] > 0 & df['curve_diff'].shift(1) == 0,'new'] = 1.0
df.loc[df['curve_diff'] < 0 & df['curve_diff'].shift(1) == 0,'new'] = -1.0
df.loc[df['new'] != 1.0 | df['new'] != -1.0,'new'] = 0

这些命令不起作用并出现此错误

TypeError: 无法将 dtyped [float64] 数组与类型 [bool] 的标量进行比较。

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。