Сохраните несколько массивов в одном файле

Question

Я хочу сохранить разные массивы в одном файле:

a = [[1,2],[3,4],[5,6]]
b = [1,3,5]

Я прочитал эту документацию о функции np.savetxt

Тем не менее, я не могу сохранить a и b в одном файле для доступа к ним в другой программе.

Я бы хотел что-то вроде этого:

a = load("file_path",a)
b = load("file_path",b)

Как я могу это сделать?

Answer 1

Вы можете использовать np.savez вместо

np.savez("file_path.npz", a=a, b=b)

А затем загрузить с

npzfile = np.load(outfile)

a = npzfile['a']
b = npzfile['b']

РЕДАКТИРОВАТЬ: Обновлен np.savez вызов, поэтому массивы сохраняются под своими именами.