【发布时间】：2014-11-09 17:09:13
【问题描述】：
我正在使用一维 numpy 数组，首先进行一些数学运算，然后将所有内容保存到单个 csv 文件中。数据集通常具有不同的长度，我无法将它们拼合在一起。这是我能想到的最好的方法，但必须有更优雅的方法。
import numpy as np
import pandas as pd
import os
array1 = np.linspace(1,20,10)
array2 = np.linspace(12,230,10)
array3 = np.linspace(7,82,20)
array4 = np.linspace(6,55,20)
output1 = np.column_stack((array1.flatten(),array2.flatten())) #saving first array set to file
np.savetxt("tempfile1.csv", output1, delimiter=',')
output2 = np.column_stack((array3.flatten(),array4.flatten())) # doing it again second array
np.savetxt("tempfile2.csv", output2, delimiter=',')
a = pd.read_csv('tempfile1.csv') # use pandas to read both files
b = pd.read_csv("tempfile2.csv")
merged = b.join(a, rsuffix='*') # merge with panda for single file
os.remove('tempfile1.csv')
os.remove("tempfile2.csv") # delete temp files
merged.to_csv('savefile.csv', index=False) # save merged file
【问题讨论】：
标签： python arrays csv numpy pandas