将多列数据放入一个数组python中

时间:2015-04-29 13:55:04

标签: python numpy genfromtxt

所以我有一组数据,其中包含x数据的单个列和y数据的多个列。

x_title     y_title_1    y_title_2   y_title_3 ....   y_title_n
data_x1     data_y2      data_y3     data_y4          data_yn
....        ....         ....        ....      ....   ....

我正在尝试在同一图表上绘制数据,每个y_data_i的不同图表。

我正在使用numpy数组,matplotlib和scipy。并且genfromtxt将其读入。 我想最好的方法是将每列放入一个数组,因此有一个y_data_1,y_data_2等数组。但是我不知道怎么做? 我知道如何单独阅读它们到一组数据,所以:

y_data_1=data[0:number_of_rows,1] 

但是,我不知道如何编写数百列的方法来编写代码。

2 个答案:

答案 0 :(得分:0)

基本的绘图功能是:

import matplotlib.pyplot as plt
x = data[0]
for y in data[1:]:
    plt.plot(x, y)
plt.show()

其中data [0]是您的第一列,data [1:]是除第一列之外的所有内容。

有关更多选项,请参阅here

答案 1 :(得分:0)

>>>import pandas as pd
>>>create a new series
>>>ser1 = pd.Series()
>>>col = [x for x in df.columns]
   df is values of csv data 
>>>for i in range(N):
   N is no of columns
   >>>ser1 = ser1.append(df[i])