pandas可以使用列作为索引吗?

时间:2016-07-23 13:39:10

标签: python excel pandas

我有一个这样的电子表格:

Locality    2005    2006    2007    2008    2009

ABBOTSFORD  427000  448000  602500  600000  638500
ABERFELDIE  534000  600000  735000  710000  775000
AIREYS INLET459000  440000  430000  517500  512500

我不想手动将列与行交换。是否可以使用pandas将数据读取到列表中:

data['ABBOTSFORD']=[427000,448000,602500,600000,638500]
data['ABERFELDIE']=[534000,600000,735000,710000,775000]
data['AIREYS INLET']=[459000,440000,430000,517500,512500]

4 个答案:

答案 0 :(得分:136)

是的,使用set_index,您可以将Locality设为行索引。

data.set_index('Locality', inplace=True)

如果未提供inplace=True,则set_index会返回修改后的数据框。

示例:

> import pandas as pd
> df = pd.DataFrame([['ABBOTSFORD', 427000, 448000],
                     ['ABERFELDIE', 534000, 600000]],
                    columns=['Locality', 2005, 2006])

> df
     Locality    2005    2006
0  ABBOTSFORD  427000  448000
1  ABERFELDIE  534000  600000

> df.set_index('Locality', inplace=True)
> df
              2005    2006
Locality                  
ABBOTSFORD  427000  448000
ABERFELDIE  534000  600000

> df.loc['ABBOTSFORD']
2005    427000
2006    448000
Name: ABBOTSFORD, dtype: int64

> df.loc['ABBOTSFORD'][2005]
427000

> df.loc['ABBOTSFORD'].values
array([427000, 448000])

> df.loc['ABBOTSFORD'].tolist()
[427000, 448000]

答案 1 :(得分:10)

您可以使用set_index按照说明更改索引。 您不需要手动将行与列交换,pandas中有一个转置(data.T)方法可以为您完成:

> df = pd.DataFrame([['ABBOTSFORD', 427000, 448000],
                    ['ABERFELDIE', 534000, 600000]],
                    columns=['Locality', 2005, 2006])

> newdf = df.set_index('Locality').T
> newdf

Locality    ABBOTSFORD  ABERFELDIE
2005        427000      534000
2006        448000      600000

然后您可以获取数据帧列值并将其转换为列表:

> newdf['ABBOTSFORD'].values.tolist()

[427000, 448000]

答案 2 :(得分:1)

另一种简单的方法是将列分配给数据框索引

data = {
  'Locality': ['ABBOTSFORD', 'ABERFELDIE', 'AIREYS INLET'],
  '2005': [427000, 534000, 459000 ],
  '2006': [448000, 448000, 448000],
  '2007': [602500, 602500, 602500],
  '2008': [600000, 710000, 517500],
  '2009': [638500, 775000, 512500]
}

df = pd.DataFrame(data)

# set the locality column as the index
df.index = df['Locality']

如果您不再需要 Locality 列作为列,您可以将其删除

df.drop('Locality', axis=1)

你会得到


              | 2005     | 2006   | 2007   | 2008   | 2009
Locality      |-------------------------------------------              
ABBOTSFORD    | 427000   | 448000 | 602500 | 600000 | 638500
ABERFELDIE    | 534000   | 448000 | 602500 | 710000 | 775000
AIREYS INLET  | 459000   | 448000 | 602500 | 517500 | 512500

答案 3 :(得分:0)

您可以在从Pandas中的电子表格读取数据时使用可用的 index_col 参数设置列索引。

这是我的解决方法:

  1. 首先,将熊猫作为pd导入: import pandas as pd

  2. 使用 pd.read_excel()(如果您的电子表格中有数据)读入文件名,并通过指定index_col参数将索引设置为“本地性”。

    df = pd.read_excel('testexcel.xlsx', index_col=0)

    在此阶段,如果出现“没有名为xlrd的模块”错误,请使用pip install xlrd安装它。

  3. 要进行目视检查,请使用df.head()读取数据框,这将打印以下输出 sc

  4. 现在,您可以获取数据框所需列的值并进行打印

    sc2