我无法在不修改它们的情况下重塑大小为 3 的 np arrays 的一维 np 数组

Question

rgb_list = []

int_list = [1, 0, 0, 1, 0, 0, 1, 1, 0, 1, 0, 0, 1, 1, 1, 0, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 0, 1, 1, 0, 1]

for num in range(0, len(int_list)-3, 3):
    rgb_list.append(received_int[num:num+3])

received_array = np.array(rgb_list)
print(received_array)

received_array_2d = np.ndarray.reshape(received_array, (5, 2))
print(received_array_2d)

So up until received_array, everything was fine, but when I try to reshape it into a 2D array, I get an error code, I assume it's because numpy is considering each integer individually, not the arrays.

ValueError: cannot reshape array of size 30 into shape (5,2)

the output of print(received_array) is

[[1 0 0]
 [1 0 0]
 [1 1 0]
 [1 0 0]
 [1 1 1]
 [0 0 1]
 [0 1 0]
 [1 0 1]
 [0 1 0]
 [0 1 1]]

I want to get a 2D array that resembles this

[[1 0 0] [1 0 0] [1 1 0] [1 0 0] [1 1 1]
 [0 0 1] [0 1 0] [1 0 1] [0 1 0] [0 1 1]]

How would I go about doing that?

Answer 1

If you are using numpy arrays, use numpy methods: reshape is appropriate here.

You first need to trim your array to a multiple of the expected dimensions:

int_list = np.array([1, 0, 0, 1, 0, 0, 1, 1, 0, 1, 0, 0, 1, 1, 1, 0, 0, 1, 0, 1, 0, 1, 0, 1, 0, 1, 0, 0, 1, 1, 0, 1])
X,Y,Z = 2,5,3
int_list[:X*Y*Z].reshape((2,5,3))

output:

array([[[1, 0, 0], [1, 0, 0], [1, 1, 0], [1, 0, 0], [1, 1, 1]],
       [[0, 0, 1], [0, 1, 0], [1, 0, 1], [0, 1, 0], [0, 1, 1]],
      ])