python - 如何截断一个numpy数组？

Question

我正在尝试使用以下代码行将“数据”(大小为 112943)截断为形状(1,15000):

data = np.reshape(data, (1, 15000))

但是，这给了我以下错误:

ValueError: cannot reshape array of size 112943 into shape (1,15000)

有关如何修复此错误的任何建议？

Answer 1

换句话说，由于您只需要前 15K 个元素，您可以为此使用基本切片:

In [114]: arr = np.random.randn(112943)

In [115]: truncated_arr = arr[:15000]

In [116]: truncated_arr.shape
Out[116]: (15000,)

In [117]: truncated_arr = truncated_arr[None, :]

In [118]: truncated_arr.shape
Out[118]: (1, 15000)