Я использую пакет ProDy на Python для извлечения координат из файлов PDB, и у меня есть список с двумя файлами .npz. Я пытаюсь получить массив для каждого файла, который содержит только 3 столбца из файла, и добавить 1 созданный мной столбец. Очевидно, это работает в исходном файле, где у меня есть переменная «xyzfile», в которой хранится массив для каждого файла. Но когда я пытаюсь импортировать переменную в другой файл, она импортирует только последний массив. Это код, который я написал:
Код: Выделить всё
os.chdir("pocket_files")
files = os.listdir()
print(files)
['1b7fH.ag.npz', '1asyH.ag.npz']
for i in files:
var = np.load(i)
print(var)
coord = var['coordinates']
coord = coord.reshape(-1, 3)
radius = np.array([1.5])
radii = np.repeat(radius, len(coord))
radii = np.expand_dims(radii, axis=1)
xyzfile = np.concatenate((coord, radii), axis=1)
print(xyzfile.shape)
И вывод print(xyzfile.shape) равен (618, 4) (1527, 4), но когда я импортирую его в другой файл:
Код: Выделить всё
from xyzrfile import xyzfile
print(xyzfile.shape)
(1527, 4)
Он печатает только форму последнего массива. Мне нужно иметь возможность хранить каждый массив в отдельной переменной, чтобы я также мог создать отдельный файл для каждого массива.
Мобильная версия