熊猫数据框

Question

我有一个数据框df，具有两列“电压”（v）和“电流”（I）。我想从文件中随机选择5个“电压”值，将其保存在[v1，v2，v3，v4，v5]等一维数组中，并将相应的电流值保存在[I1，I2， ...，I5]。这是我尝试过的：

 df=pd.read_csv(file,sep=",",header=None,usecols=[0,1],names=['voltage','current'])

#pick 5 random values of voltage and save it in np array
 V= np.array( df['voltage'].sample(n=5))

如何在选定的V值下对相应的I值执行相同的操作？

Answer 1

我认为需要：

arr = df.sample(n=5).values
a = arr[:, 0]
b = arr[:, 1]

Answer 2

尽管jezrael的答案确实提供了所需的输出，但您的问题的答案将是：

V= df['voltage'].sample(n=5)
I = df.loc[V.index,'current']