Iva*_*tch 4 python performance sum pandas
我有两个数组:
array1 = [1,2,3]
array2 = [10,20,30]
Run Code Online (Sandbox Code Playgroud)
我想要下一笔钱:
array3 = [10+1,10+2,10+3,20+1,20+2,20+3,30+1,30+2,30+3]
Run Code Online (Sandbox Code Playgroud)
我怎样才能做到这一点?(我知道可以通过两个for循环来完成,但如果可能的话我想要更有效的东西)
注意:这两个数组包含在数据帧(pandas)中
我不认为这里有大熊猫
[x+y for x in array2 for y in array1]
Out[293]: [11, 12, 13, 21, 22, 23, 31, 32, 33]
Run Code Online (Sandbox Code Playgroud)
如果它们在数据框中
df=pd.DataFrame({'a':array1,'b':array2})
df
Out[296]:
a b
0 1 10
1 2 20
2 3 30
df.a.values+df.b.values[:,None]
Out[297]:
array([[11, 12, 13],
[21, 22, 23],
[31, 32, 33]], dtype=int64)
Run Code Online (Sandbox Code Playgroud)
更新
(df.a.values+df.b.values[:,None]).ravel()
Out[308]: array([11, 12, 13, 21, 22, 23, 31, 32, 33], dtype=int64)
Run Code Online (Sandbox Code Playgroud)
| 归档时间: |
|
| 查看次数: |
68 次 |
| 最近记录: |