使用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
