行の取得: .iloc[] と .loc[]

pandasで効率的にコードを書く

Leonidas Souliotis

PhD Candidate

ポーカーデータセット

	S1	R1	S2	R2	S3	R3	S4	R4	S5	R5
1	♦	10	♣	Jack	♣	King	♠	4	♥	Ace
2	♦	Jack	♦	King	♦	10	♦	Queen	♦	Ace
3	♣	Queen	♣	Jack	♣	King	♣	10	♣	Ace

	S1	R1	S2	R2	S3	R3	S4	R4	S5	R5
1	2	10	3	11	3	13	4	4	1	1
2	2	11	2	13	2	10	2	12	2	1
3	3	12	3	11	3	13	3	10	3	1

Sn: n枚目のカードのスート

1 (ハート), 2 (ダイヤ), 3 (クラブ), 4 (スペード)

Rn: n枚目のカードのランク

1 (エース), 2–10, 11 (ジャック), 12 (クイーン), 13 (キング)

狙った行を取得

.loc[] — インデックス名で取得

# 取得する行範囲を指定
rows = range(0, 500)
# .loc[] の計測
loc_start_time = time.time()
data.loc[rows]    
loc_end_time = time.time()

print("Time using .loc[] : {} sec".format(
            loc_end_time - loc_start_time))

Time using .loc[]: 0.001951932 seconds

.iloc[] — インデックス番号で取得

# 取得する行範囲を指定
rows = range(0, 500)
# .iloc[] の計測
iloc_start_time = time.time()
data.iloc[rows]    
iloc_end_time = time.time()

print("Time using .iloc[]: {} sec".format(
            iloc_end_time - iloc_start_time)

Time using .iloc[] : 0.0007140636 sec

Difference in speed: 173.355592654%

狙った列を取得

.iloc[] — インデックス番号で取得

iloc_start_time = time.time()
data.iloc[:,:3]
iloc_end_time = time.time()
print("Time using .iloc[]: {} sec".format(
            iloc_end_time - iloc_start_time))

Time using .iloc[]: 0.00125193595886 sec

列名で取得

names_start_time = time.time()
data[['S1', 'R1', 'S2']]
names_end_time = time.time()
print("Time using selection by name: {} sec".format(
            names_end_time - names_start_time))

Time using selection by name: 0.000964879989624 sec

Difference in speed: 29.7504324188%

やってみましょう！

pandasで効率的にコードを書く