博客

如何在Python-Pandas中把数据框架列转换成索引

  • Post category:Python

在Python-Pandas中，我们可以通过使用set_index()方法，将数据框架中的某一列（或多列）转换成索引。下面是这一过程的详细步骤：

  1. 引入Pandas库并读取数据

首先，我们需要引入Pandas库并读取我们需要转换的数据框架。这里举一个例子，假设我们有这样一个包含“年份”、“季度”和“销售额”三列数据的数据框架data：

import pandas as pd

data = pd.DataFrame({
    'year': [2018, 2018, 2019, 2019],
    'quarter': ['Q1', 'Q2', 'Q1', 'Q2'],
    'sales': [100, 200, 300, 400]
})
  1. 将列转换成索引

接下来，我们可以使用set_index()方法将“年份”和“季度”这两列转换成组合索引。代码如下：

data.set_index(['year', 'quarter'], inplace=True)

注意，set_index()方法的第一个参数传递列名的列表，意味着你可以传递多个列进行索引转换。inplace=True参数用来表示直接在原始数据框架上对其进行修改。

  1. 查看结果

现在，我们已经将 “year” 和 “quarter” 这两列转换成了组合索引，我们可以使用 Pandas 的相关方法对数据进行操作。例如，我们可以使用 loc 方法选择特定的索引行/列数据:

data.loc[(2018, 'Q1'): (2019, 'Q2'), :]

这样我们就可以看到仅包含“销售额”列的数据框架，并且其行索引为组合索引 “年份”、“季度”：

              sales
year quarter       
2018 Q1         100
     Q2         200
2019 Q1         300
     Q2         400

以上就是在Python-Pandas中把数据框架列转换成索引的完整攻略。

Tags: