Python сохранить массив NumPy только с ненулевыми элементами

Question

У меня есть массив, большинство элементов которого равны нулю.

A =  [ 1,0,2
       2,3,0
       0,0,4 ]

Я хочу сохранить это как

rowid[0] colid[0] 1
rowid[0] colid[2] 2
rowid[1] colid[0] 2
rowid[1] colid[1] 3
rowid[2] colid[2] 4

здесь rowid и colid - это массивы, которые отображают индексы массивафактические записи в исходном файле.

Как я могу сделать это без использования цикла for?

Answer 1

A = np.array(A).reshape(3, 3) # make A a 3x3 numpy array 
i, j = np.where(A != 0) # find indices where it is nonzero 
v = A[i, j] # extract nonzero values of the array 
np.savetxt('file.csv', np.vstack((i, j, v)).T, delimiter = ',') # stack and save 

# @Daniel F suggestion is to make header with array shape and add delimiter kwarg
np.savetxt('file.csv', np.vstack((i, j, v)).T, delimiter = ',', header = str(A.shape))