pandasのデータを取り出すとデータ型が勝手に変わってしまう

Question

前提・実現したいこと

pandasデータフレームからlocメソッドでデータを取り出すと、int型が勝手にfloat型になってしまうのを防ぎたいです。
アドバイスお願いします。

発生していること

行のデータをリストに保存しようとしています。

column(rc, ip_bc, ip_ec)はint型でキープしたいのですが、
locで取り出したデータは全てfloat型になってしまいます。

df3
　　　　rc	ip_bc	ip_ec	xc	yc
0	2	1	2	40.0	3.0

In: rowlist = df3.loc[0, ['rc', 'ip_bc', 'ip_ec', 'xc', 'yc']].values.tolist()

Out: [2.0, 1.0, 2.0, 40.0, 3.0]

補足

python 3.7.1
pandas 0.24.2

Answer 1

Series は1つの型しかとれないので、行を loc で取得した際に int と float など異なる型が混在する場合はより広い型にキャストされてしまいます。

import pandas as pd

df = pd.DataFrame({"a": [1, 2, 3], "b": [1.0, 2.0, 3.0]})

print(df)
#    a    b
# 0  1  1.0
# 1  2  2.0
# 2  3  3.0

# float になってしまう
print(df.loc[0])
# a    1.0
# b    1.0
# Name: 0, dtype: float64

itertuples() で DataFrame を Namedtuple のリストに変換してから、インデックスで取得するとよいかと思います。

import pandas as pd

df = pd.DataFrame({"a": [1, 2, 3], "b": [1.0, 2.0, 3.0]})

# DataFrame をタプルのリストに変換する。
lst = [t for t in df.itertuples()]

# 以下のようにアクセスする。元の型は保存されている。
print(lst[0])  # Pandas(Index=0, a=1, b=1.0)
print(lst[0].a, lst[1].b)  # 1 2.0

---

不明な点があれば、コメントしてください。

Comments

[takashim]

どうもありがとうございます。毎度助かります。

最終的に欲しい形はリストのリストで、頂いた例で表現すると
[[1, 1.0], [2, 2.0], [3, 3.0]] です。

ここまで持っていけますでしょうか。
今、tolist()とか試してみましたがうまくいきません。

[takashim]

for i in lst:
print(list(i[1: 3]))
こんな感じで何とか希望の形までもってこれました。
ありがとうございました。