2017-04-21 29 views
0

请考虑以下程序,其中,我创建了包含三列的多索引数据框,并最终使用列表元组的列表的嵌套列表填充了一列。我把这些索引弄平了,并试着迭代行ix, rec = next(df.iterrows())。 然后我从迭代的行(rec)取消参考数据列rec.data,发现它是一个内存对象<memory at 0x000000000D6E0AC8>。在调用记录rec.data.obj上的obj时,我意识到它是一个包含整行内容的数组。要获得实际内容,我必须获取非常不直观的项目索引。数据帧列向行的递归引用

>>> print(rec.data.obj[2]) 
[(['9', '"', 'X', '12', '"'], 0.9993008259451988)] 

样品重新创建实例

def foo(): 
    return [(['9', '"', 'X', '12', '"'], 0.99930082594519876)] 
import pandas as pd 

def spam(): 
    index = pd.MultiIndex(levels=[[], []], 
          labels=[[], []], 
          names=[u'timestamp', u'key']) 
    columns = ['data', 'col1', 'col2'] 
    df = pd.DataFrame(index=index, columns=columns) 
    for ix in range(4): 
     key = ('XXX', ix) 
     df.loc[key, 'data'] = str(foo()) 
     df.loc[key, 'col1'] = "col1_{}".format(ix) 
     df.loc[key, 'col2'] = "col2_{}".format(ix) 
    df.reset_index(inplace=True) 
    return df 

def bar(): 
    df = spam() 
    ix, rec = next(df.iterrows()) 
    print(rec.data) 
    print(rec.data.obj) 
    print(rec.data.obj[2]) 

bar() 

输出

<memory at 0x000000000D6E0AC8> 
['XXX' 0 '[([\'9\', \'"\', \'X\', \'12\', \'"\'], 0.9993008259451988)]' 
'col1_0' 'col2_0'] 
[(['9', '"', 'X', '12', '"'], 0.9993008259451988)] 

我不知所云,无法理解,我缺少什么

回答

1

看来你需要itertuples

def bar(): 
    df = spam() 
    rec = next(df.itertuples()) 
    print (rec) 
    print (rec.data) 
bar() 
Pandas(Index=0, timestamp='XXX', 
     key=0, 
     data='[([\'9\', \'"\', \'X\', \'12\', \'"\'], 0.9993008259451988)]', 
     col1='col1_0', 
     col2='col2_0') 

[(['9', '"', 'X', '12', '"'], 0.9993008259451988)] 
+0

谢谢。我正在使用'iterrows',因为我想避免通过'itertuples'调用enumerate。无论如何,任何想法为什么我的代码失败? – Abhijit

+0

不,我有想法。也许对于迭代器可能使用'print(rec.Index)' – jezrael