【Python・pandas】DataFrame から index を指定して1行取得したい

Question

いつもお世話になっております。

早速本題に入らせていただきます。

現在、pandas.DataFrame から index を指定して1行のみ取得しようと試みているのですが、
下記のエラーが出てしまい取得できませんでした。

KeyError: '1'

使用しているDataFrameの中身は下記です。

index	1	2	3
1	101	201	301
2	102	202	302
3	103	203	303
4	104	204	304
5	105	205	305

---

現在の取得している部分のコードは下記です。

df_users.loc[user_id]

---

想定している取得結果は下記となります。

index = 2 を指定したとき　下記の1行のみ取得したいです

index	1	2	3
2	102	202	302

---

お時間ある方で、indexを指定し1行取得する方法をご存じの方いらっしゃいましたらご助力頂けますと幸いです。

---

追記

print(df_users.index)

Int64Index([  1,   2,   3,   4,   5], dtype='int64', length=5)

Comments

[tiitoi]

print(df_users.index) とすると、どのように出力されますか？

[pekopekocot]

ありがとうございます、質問文に追記いたしました。

Answer 1

Int64Index([ 1, 2, 3, 4, 5], dtype='int64', length=5) となっているのであれば、df_users.loc[2] で取得できると思います。
df_users.loc[user_id] で Key Error となるということは、df_users.index の中に user_id が存在しないことを意味します。

サンプルコード

import pandas as pd


df = pd.DataFrame(
    [
        [101, 201, 301],
        [102, 202, 302],
        [103, 203, 303],
        [104, 204, 304],
        [105, 205, 305],
    ],
    index=[1, 2, 3, 4, 5],
    columns=[1, 2, 3]
)
print(df)
#      1    2    3
# 1  101  201  301
# 2  102  202  302
# 3  103  203  303
# 4  104  204  304
# 5  105  205  305

print(df.index)
# Int64Index([1, 2, 3, 4, 5], dtype='int64')

# 1行だけ Series として取得したい場合
print(df.loc[2])
# 1行分DataFrame として取得したい場合
print(df.loc[2:2])
#      0    1    2
# 2  102  202  302

追記

考えられる可能性として、index という列名が DataFrame のインデックスになっていない可能性があります。その場合、set_index() でインデックスに設定できます。

import pandas as pd


df = pd.DataFrame(
    [
        [1, 101, 201, 301],
        [2, 102, 202, 302],
        [3, 103, 203, 303],
        [4, 104, 204, 304],
        [5, 105, 205, 305],
    ],
    columns=["index", 1, 2, 3]
)
print(df)
#    index    1    2    3
# 0      1  101  201  301
# 1      2  102  202  302
# 2      3  103  203  303
# 3      4  104  204  304
# 4      5  105  205  305

df = df.set_index("index")
print(df.loc[2])

Comments

[pekopekocot]

回答ありがとうございました。
ご指摘の方法でも取得できなかったので型を調べてみたところ user_id が string型になっていました...
intにキャストしたところ、無事に1行取得することができました。