numpy 可变矩阵的np数组

643ylb08  于 2023-02-04  发布在  其他
关注(0)|答案(1)|浏览(164)
import numpy as np

 data = np.array([[10, 20, 30, 40, 50, 60, 70, 80, 90],
    [2, 7, 8, 9, 10, 11],
    [3, 12, 13, 14, 15, 16],
    [4, 3, 4, 5, 6, 7, 10, 12]],dtype=object)

target = data[:,0]

它有这个错误。

IndexError     Traceback (most recent call last)

       Input In \[82\], in \<cell line: 9\>()

 data =  np.array(\[\[10, 20, 30, 40, 50, 60, 70, 80, 90\],

       \[2, 7, 8, 9, 10, 11\],
\[3, 12, 13, 14, 15, 16\],
\[4, 3, 4, 5, 6, 7, 10,12\]\],dtype=object)

  # Define the target data ----\> 9 target = data\[:,0\]

   IndexError: too many indices for array: array is 1-dimensional, but 2 were indexed

我可以知道如何修复它吗?我的意思是不要改变数据中的元素。非常感谢。我做了相同大小的矩阵,错误信息消失了。但我有可变大小的数据。

a8jjtwal

a8jjtwal1#

你有一个对象数组,所以你不能在axis=1上使用索引,因为没有(data.shape-〉(4,))。
使用列表解析:

out = np.array([a[0] for a in data])

输出:array([10, 2, 3, 4])

相关问题