為 arrays 列表賦值

Question

我是 numpy 的新手，想知道如何為數組列表賦值。 在 C++ 我可以做array[i][j] = 1我怎樣才能為 numpy 做到這一點？ 例如，如果我有： array = np.array([[0,0], [0, 1], [0, 2]])我如何分配[0, 0]以具有例如1的值?

謝謝你們

Answer 1

list_of_arrays不是變量的好名稱，因為它是 2d numpy 數組而不是列表。 也就是說，實際上您需要做的就是list_of_arrays[0, 0] = 1 。 您不應該嘗試像 C++ 中那樣嘗試list_of_arrays[0][0] = 1 - 它會起作用，但效率會低得多，因為它為子陣列構造了一個臨時的 object。

>>> import numpy as np

>>> list_of_arrays = np.array([[0,0], [0, 1], [0, 2]])

>>> list_of_arrays
array([[0, 0],
       [0, 1],
       [0, 2]])

>>> list_of_arrays[0, 0] = 1

>>> list_of_arrays
array([[1, 0],
       [0, 1],
       [0, 2]])