Pandas選擇一行暗藏玄機

本來以為pandas通過loc然后指定一個index,就可以正確返回一行,沒想到其實還暗藏玄機,我們來看看如何正確的獲取pandas的一行:
首先我們創建一個pandas的DataFrame:

import pandas as pd
import numpy as np
df = pd.DataFrame(np.arange(0,16).reshape(4,4))
df

我們按照正常的思維拿到DataFrame的第一行并打印一下輸出結果類型:

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,那我們如何才能獲得一個DataFrame類型的一行數據呢,多加一個中括號就可以啦:

df.loc[[0]]

輸出如下:

    0   1   2   3
0   0   1   2   3
最后編輯于
?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。

推薦閱讀更多精彩內容