pandas DataFrame 行列索引及值的获取的方法
本文详解 pandas DataFrame 的行列索引创建与值提取方法。通过 index 和 columns 参数自定义索引,演示如何获取单列、单行及特定单元格数值。结合车站客流数据实例,展示利用 iloc 和 loc 进行数据筛选与统计计算,如查找最大客流车站及其日均值对比。
DataFrame 索引基础与创建
pandas DataFrame 作为二维数据结构,同时具备列索引与行索引功能。
前文仅涉及列索引的讨论,此处补充说明。
import pandas as pd
df = pd.DataFrame({'A': [0, 1, 2], 'B': [3, 4, 5]})
print df
# 结果:
A B
0 0 3
1 1 4
2 2 5
若未显式指定,行索引将自动生成从 0 开始的整数序列,例如 0, 1, 2。
若需自定义行索引和列索引,可分别通过 index 和 columns 参数进行设定。
以下示例数据展示了 5 个车站在 10 天内的客流统计信息。
ridership_df = pd.DataFrame(
data=[[ 0, 0, 2, 5, 0],
[1478, 3877, 3674, 2328, 2539],
[1613, 4088, 3991, 6461, 2691],
[1560, 3392, 3826, 4787, 2613],
[1608, 4802, 3932, 4477, 2705],
[1576, 3933, 3909, 4979, 2685],
[ 95, 229, 255, 496, 201],
[ 2, 0, 1, 27, 0],
[1438, 3785, 3589, 4174, 2215],
[1342, 4043, 4009, 4665, 3033]],
index=['05-01-11', '05-02-11', '05-03-11', '05-04-11', '05-05-11',
'05-06-11', '05-07-11', '05-08-11', '05-09-11', '05-10-11'],
columns=['R003', 'R004', 'R005', 'R006', 'R007']
)
其中 data 参数接收一个 numpy 二维数组,index 参数用于定义行索引,columns 参数用于定义列索引。
生成的 DataFrame 以表格形式呈现,便于直观查看数据分布。
R003 R004 R005 R006 R007
05-01-11 0 0 2 5 0
05-02-11 1478 3877 3674 2328 2539
05-03-11 1613 4088 3991 6461 2691
05-04-11 1560 3392 3826 4787 2613
05-05-11 1608 4802 3932 4477 2705
05-06-11 1576 3933 3909 4979 2685
05-07-11 95 229 255 496 201
05-08-11 2 0 1 27 0
05-09-11 1438 3785 3589 4174 2215
05-10-11 1342 4043 4009 4665 3033
DataFrame 值的获取方法
接下来介绍如何从 DataFrame 中提取具体数值。
1. 获取某一列数据:直接使用列名作为键,即 ['key']。
print(ridership_df['R003'])
# 结果:
05-01-11 0
05-02-11 1478
05-03-11 1613
05-04-11 1560
05-05-11 1608
05-06-11 1576
05-07-11 95
05-08-11 2
05-09-11 1438
05-10-11 1342
Name: R003, dtype: int64
2. 获取某一行数据:使用 .loc['key'] 方法。
print(ridership_df.loc['05-01-11'])
# 或者
print(ridership_df.iloc[0])
# 结果:
R003 0
R004 0
R005 2
R006 5
R007 0
Name: 05-01-11, dtype: int64
3. 获取特定行与特定列交叉处的单个值:
print(ridership_df.loc['05-05-11','R003'])
# 或者
print(ridership_df.iloc[4,0])
# 结果:
1608
4. 获取底层原始的 numpy 二维数组:
print(ridership_df.values)
# 结果:
[[ 0 0 2 5 0]
[1478 3877 3674 2328 2539]
[1613 4088 3991 6461 2691]
[1560 3392 3826 4787 2613]
[1608 4802 3932 4477 2705]
[1576 3933 3909 4979 2685]
[ 95 229 255 496 201]
[ 2 0 1 27 0]
[1438 3785 3589 4174 2215]
[1342 4043 4009 4665 3033]]
*注意:在此过程中,若数据格式不一致,pandas 会自动进行类型转换。
综合应用示例
下面提供一个综合应用示例。
从 ridership_df 中找出第一天客流量最大的车站,返回该车站的日平均客流,并返回所有车站的平均日客流作为对比参考。
def mean_riders_for_max_station(ridership):
max_index = ridership.iloc[0].argmax()
mean_for_max = ridership[max_index].mean()
overall_mean = ridership.values.mean()
return (overall_mean, mean_for_max)
print mean_riders_for_max_station(ridership_df)
# 结果:
(2342.6, 3239.9)
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持本站。
本站声明
本文内容仅供参考,不代表本站观点。转载或引用请注明出处。如发现内容问题,请通过联系我们页面反馈。













