我对numpy和pandas相对较新(我是一名实验性物理学家,所以我多年来一直使用ROOT ......).ROOT中的常见图是2D散点图,其中给定x和y值的列表,得到一个变量与另一个变量的"热图"型散点图.
如何用numpy和Pandas最好地完成这项工作？我正在尝试使用该Dataframe.plot()功能,但我甚至都在努力创建Dataframe.
import numpy as np
import pandas as pd
x = np.random.randn(1,5)
y = np.sin(x)
df = pd.DataFrame(d)
首先,这个数据框有形状(1,2),但我希望它有形状(5,2).如果我能够使数据帧具有正确的形状,我相信我可以找出DataFrame.plot()绘制我想要的功能.
我有两个python列表,
A = [ 1, 2, 3, 4, 5 ]
B = [ True, False, False, True, True ]
列表A和B的长度相同.
我想只总结A中与B中的True元素相对应的元素.我知道我可以用以下方法做到:
sum([A[x] for x in xrange(len(A)) if B[x]])
但我想知道是否有一个更优雅的解决方案,不涉及循环每个列表中的元素？