按索引和列名数组分割Pandas数据帧

2024-04-20 15:08:26 发布

您现在位置:Python中文网/ 问答频道 /正文

我希望用pandas数据帧复制numpy数组的行为。我想传递一个索引和列名数组,并获取在相应的索引和列名中找到的对象列表。

import pandas as pd
import numpy as np

在纽比:

array=np.array(range(9)).reshape([3,3])
print array
print array[[0,1],[0,1]]

[[0 1 2]
 [3 4 5]
 [6 7 8]]

[0 4]

在熊猫中:

prng = pd.period_range('1/1/2011', '1/1/2013', freq='A')
df=pd.DataFrame(array,index=prng)
print df

      0  1  2
2011  0  1  2
2012  3  4  5
2013  6  7  8

df[[2011,2012],[0,1]]

预期产量:

[0 4]

如何分割此数据帧以使其返回与numpy相同的值?


Tags: 数据对象importnumpypandasdf列表as