Сохраните множество переменных в одном файле

Question

У меня есть скрипт, который собирает данные, используя np.loadtxt() из разных папок.Я хотел бы иметь возможность сохранить все из них в текущей папке без необходимости делать savetxt() для каждого.

Цель состоит в том, чтобы иметь переносимый файл для выполнения сценария на другом компьютере.

Answer 1

Вы можете использовать функцию NumPy save следующим образом:

import numpy as np
# Create sample data
a = np.array([1.0, 2.0, 3.0])
b = np.array([4.0, 5.0])
# Save to file
np.save('myfile.npy', [a, b])
# Load back in
a2, b2 = np.load('myfile.npy')

Обратите внимание, что хотя в документации это не указано явно, вы действительно можете хранить несколько массивов разных размеров.в одном файле, как указано выше (на самом деле это делается путем сохранения каждого массива как элемента одномерного массива dtype object, но не обращайте внимания на реализацию).