pandas - 两个熊猫数据帧的条件元素乘法
问题描述
我正在努力在两个数据帧之间进行基本的条件元素乘法。假设我有以下两个数据框:
df1 = pd.DataFrame({'A': [-0.1,0.3,-0.4, 0.8,-0.5,-0.1,0.3,-0.4, 0.8,-1.2],'B': [-0.2,0.5,0.3,-0.5,0.1,-0.2,0.5,0.3,-0.5,0.9]},index=[0, 1, 2, 3,4,5,6,7,8,9])
df2=pd.DataFrame({'C': [-0.003,0.03848,-0.04404, 0.018,-0.1515,-0.02181,0.233,-0.0044, 0.01458,-0.015],'D': [-0.0152,0.0155,0.03,-0.0155,0.0151,-0.012,0.035,0.0013,-0.0005,0.009]},index=[0, 1, 2, 3,4,5,6,7,8,9])
这个想法是df1 and df2.shift(-1)
根据 df1 的值进行乘法(逐元素,而不是矩阵乘法)。如果(df1>=0.50 or df1<=-0.50)
然后我将 df1 和 df2.shift(-1) 相乘。否则,我只输入 0。
此示例中所需的结果应如下所示(列名是 df1 的列名以及 df1 索引):
df3=pd.DataFrame({'A': [0,0,0, -0.1212,0.010905,0,0,0, -0.012,'NaN'],'B': [0,0.015,0,-0.00755,0,0,0.00065,0,-0.0045,'NaN']},index=[0, 1, 2, 3,4,5,6,7,8,9])
我尝试了以下代码:
import numpy as np
import pandas as pd
df3=np.where((df1>=0.50 or df1 <=-0.50),df1*df2.shift(-1),0)
我得到了 DataFrame 的真值是不明确的。使用 a.empty、a.bool()、a.item()、a.any() 或 a.all()。谢谢你。
解决方案
使用构造函数|
按OR
位DataFrame
:
arr = np.where((df1>=0.50) | (df1 <=-0.50),df1*df2.shift(-1),0)
df3 = pd.DataFrame(arr, index=df1.index, columns=df1.columns)
print (df3)
A B
0 0.000000 0.00000
1 0.000000 0.01500
2 0.000000 0.00000
3 -0.121200 -0.00755
4 0.010905 0.00000
5 0.000000 0.00000
6 0.000000 0.00065
7 0.000000 0.00000
8 -0.012000 -0.00450
9 NaN NaN
Numpy 解决方案应该更快:
arr2 = np.concatenate([df2.values[1:, ],
np.repeat(np.nan, len(df2.columns))[None, :]])
arr = np.where((df1.values>=0.50) | (df1.values <=-0.50),df1.values*arr2,0)
df3 = pd.DataFrame(arr, index=df1.index, columns=df1.columns)
print (df3)
A B
0 0.000000 0.00000
1 0.000000 0.01500
2 0.000000 0.00000
3 -0.121200 -0.00755
4 0.010905 0.00000
5 0.000000 0.00000
6 0.000000 0.00065
7 0.000000 0.00000
8 -0.012000 -0.00450
9 NaN NaN
推荐阅读
- laravel - 搜索功能,只显示两个特定的人,而不是数据库中的其他人
- c - 初始化并声明为“extern”,为什么我看到以下警告?
- python - 使用 Pillow 和 img2pdf 将图像转换为 pdf
- reactjs - 受控表单字段不遵守步骤
- autotools - AC_SEARCH_LIBS 在 gswteos-10 中找不到函数
- php - 如何在laravel中获取数组元素的文件扩展名
- python - 如何使用嵌套选项卡处理 Django 分页
- reactjs - 如何将 react-final-from 计算字段与对象数组一起使用
- c# - SNMP 如何添加额外的 OID 来请求?
- swift - Swift - SpriteKit 如何使用两个独特的背景进行无限背景滚动?