当我将.ix与DataFrame一起使用时,有没有办法强制pandas总是返回一个DataFrame?
例如,如果我运行以下行,
- import pandas as pd
- import numpy as np
- df = pd.DataFrame(np.arange(6).reshape(3,2),index=[0,1])
- x = df.ix[0]
- y = df.ix[1]
然后x将是一个DataFrame,因为0在索引中出现两次,而y将是一个Series,因为1在索引中是唯一的.我希望y也是一个DataFrame(因为我在结果上使用了iterrows(),它没有为Series定义).
我可以检查.ix返回的类型并在需要时将其转换为DataFrame,但我认为有更好的方法.
注意:截至Pandas v0.20,.ix indexer is deprecated支持.iloc / .loc.
解决方法
是的,用列表索引它:
- > x = df.ix[[0]]
- > y = df.ix[[1]]
- > type(x)
- pandas.core.frame.DataFrame
- > type(y)
- pandas.core.frame.DataFrame