本來以為pandas通過loc然后指定一個index件豌,就可以正確返回一行,沒想到其實還暗藏玄機工三,我們來看看如何正確的獲取pandas的一行:
首先我們創(chuàng)建一個pandas的DataFrame:
import pandas as pd
import numpy as np
df = pd.DataFrame(np.arange(0,16).reshape(4,4))
df
我們按照正常的思維拿到DataFrame的第一行并打印一下輸出結(jié)果類型:
df.loc[0]
print(type(df.loc[0]))
上面的輸出是:
0 0
1 1
2 2
3 3
Name: 0, dtype: int64
<class 'pandas.core.series.Series'>
他返回的是一個Series运杭,那我們?nèi)绾尾拍塬@得一個DataFrame類型的一行數(shù)據(jù)呢院塞,多加一個中括號就可以啦:
df.loc[[0]]
輸出如下:
0 1 2 3
0 0 1 2 3