pandas DataFrame 行列索引及值的获取的方法

半杯拿铁
半杯拿铁
发布时间:2026-10-13 | 4人浏览过
原创

本文详解 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)

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持本站。

本站声明

本文内容仅供参考,不代表本站观点。转载或引用请注明出处。如发现内容问题,请通过联系我们页面反馈。

Python 相关产品和应用

全部下载

专题列表