获取3D numpy数组的第一列

rkkpypqq  于 2023-06-23  发布在  其他
关注(0)|答案(1)|浏览(134)

我有两个numpys数组:

true_values = (464, 1, 1)
pred_values = (464, 1, 4))

我只想从两个数组中提取第一个尺寸值(464),因为我想比较两个数组的第一个尺寸的行。
我试着这样做:

trues_values[0,:,-1]

。但它只给我一个值,而不是464

1yjd4xko

1yjd4xko1#

正如@stef提到的,你需要以不同的方式切片你的数组。

trues_values[:, 0, -1]

每个值均表示:

  • :选择第一维中的所有元素
  • 0选择第二维中的第一个元素
  • -1选择第三维中的最后一个元素

相关问题