使用Pandas

安装

pip install pandas

DataFrame

一个DataFrame可以看成一个二维表格，就像excel一样。

import numpy as np
import pandas as pd

data = np.array([['', 'Col1', 'Col2'],
                 ['Row1', 1, 2],
                 ['Row2', 3, 4]])

df = pd.DataFrame(data=data[1:, 1:],
                  index=data[1:, 0],
                  columns=data[0, 1:])

print(df)

输出

     Col1 Col2
Row1    1    2
Row2    3    4

获取DataFrame中的值

print(df.iloc[1][1])
# 4

print(df.loc['Row2']['Col2'])
# 4

print(df.iloc[1,])
# Col1    3
# Col2    4
# Name: Row2, dtype: object

print(df.iloc[:,1])
# Row1    2
# Row2    4
# Name: Col2, dtype: object

从csv读取

data.csv

,2014,2015,2016
营业收入,1.25,1.87,2.2
营业成本,0.4,0.5,0.6
净利润,0.82,1.3,1.7

�df = pd.read_csv('data.csv', index_col=0)
print(df)

输出

      2014  2015  2016
营业收入  1.25  1.87   2.2
营业成本  0.40  0.50   0.6
净利润   0.82  1.30   1.7