使用熊猫访问合并的数据

Question

我有一组数据，已将其放入数据框，然后进行装箱：

print(data1)


[[-1.90658883e+00  5.66881290e-01  1.45443907e+00]
 [-1.82926850e+00  2.53325112e-01  1.45480072e+00]
 [-1.59073925e+00  5.33264011e-01  1.45461954e+00]
 ...
 [ 2.86246982e+02  4.52961148e-01  6.19121328e+00]]

df = pd.DataFrame(data=data1,)
print(df)
bins  = [0,50,100,150,200,250,300,400]
df1 = pd.cut(df[0],bins, labels = False)
print(df1)


1     0
2     0
 ..
500   4
501   4
502   5

0到5是bin标签。我希望能够访问每个bin /类别中的数据并将其存储在变量中。像这样：

x = df1(4) # this doesnt work, just an example.

^表示我想访问存储在pandas数据帧中第4个bin中的数据，并将其作为数组分配给变量x，但是我不确定该怎么做。

Answer 1

您可以使用pandas.DataFrame.loc并将一个布尔数组传递给它。

bi = pd.cut(df[0], bins, labels=False)
x = df.loc[bi == 4]