Записать несколько numpy массивов в файл
Я знаю, как использовать numpy.savetxt для записи массива в файл. Как я могу записать несколько массивов в один и тот же файл?
По сути, я хочу сделать математику для столбца чисел, а затем заменить старый столбец измененными числами. Я прочитал самый простой способ сделать это - написать новый файл полностью, вставить измененные числа и просто «скопировать и вставить» другие числа в файл.
Любая помощь приветствуется.
Спасибо!
1 ответ
Если вы хотите записать несколько массивов в файл для дальнейшего использования, посмотрите numpy.savez .
Однако, судя по вашему описанию, похоже, что вы хотите что-то сделать с определенным столбцом текстового файла с разделителями.
В этом случае просто загрузите все и работайте только с нужным столбцом.
Например.
import numpy as np
data = np.loadtxt('test.txt')
# Multiply the 4th column by 5
data[:,3] *= 5
# Do something more complicated to the 2nd column
data[:,1] = np.cos(data[:,1])
# Save the array back to the file
np.savetxt('test.txt', data)