DataFrame对象的创建,修改,合并
import pandas as pdimport numpy as np
创建DataFrame对象
# 创建DataFrame对象df = pd.DataFrame([1, 2, 3, 4, 5], columns=['cols'], index=['a','b','c','d','e'])print df
colsa 1b 2c 3d 4e 5
df2 = pd.DataFrame([[1, 2, 3],[4, 5, 6]], columns=['col1','col2','col3'], index=['a','b'])print df2
col1 col2 col3a 1 2 3b 4 5 6
df3 = pd.DataFrame(np.array([[1,2],[3,4]]), columns=['col1','col2'], index=['a','b'])print df3
col1 col2a 1 2b 3 4
df4 = pd.DataFrame({'col1':[1,3],'col2':[2,4]},index=['a','b'])print df4
col1 col2a 1 2b 3 4
创建DataFrame对象的数据可以为列表,数组和字典,列名和索引为列表对象
基本操作
# DataFrame对象的基本操作df2.index
Index([u'a', u'b'], dtype='object')
df2.columns
Index([u'col1', u'col2', u'col3'], dtype='object')
# 根据索引查看数据df2.loc['a'] # 索引为a这一行的数据# df2.iloc[0] 跟上面的操作等价,一个是根据索引名,一个是根据数字索引访问数据
col1 1col2 2col3 3Name: a, dtype: int64
print df2.loc[['a','b']] # 访问多行数据,索引参数为一个列表对象
col1 col2 col3a 1 2 3b 4 5 6
print df.loc[df.index[1:3]]
colsb 2c 3
# 访问列数据print df2[['col1','col3']]
col1 col3a 1 3b 4 6
计算
# DataFrame元素求和# 默认是对每列元素求和print df2.sum()
col1 5col2 7col3 9dtype: int64
# 行求和print df2.sum(1)
a 6b 15dtype: int64
# 对每个元素乘以2print df2.apply(lambda x:x*2)
col1 col2 col3a 2 4 6b 8 10 12
# 对每个元素求平方(支持ndarray一样的向量化操作)print df2**2
col1 col2 col3a 1 4 9b 16 25 36
列扩充# 对DataFrame对象进行列扩充df2['col4'] = ['cnn','rnn']print df2
col1 col2 col3 col4a 1 2 3 cnnb 4 5 6 rnn
# 也可以通过一个新的DataFrame对象来定义一个新列,索引自动对应df2['col5'] = pd.DataFrame(['MachineLearning','DeepLearning'],index=['a','b'])print df2
col1 col2 col3 col4 col5a 1 2 3 cnn MachineLearningb 4 5 6 rnn DeepLearning
新闻热点
疑难解答