python DataFrame中loc与iloc取数据的基本方法实例

1.准备一组DataFrame数据

import pandas as pd
df = pd.DataFrame({
    'AAA': [120, 101, 106, 117, 114, 122],
    'BBB': [115, 100, 110, 125, 123, 120],
    'CCC': [109, 112, 125, 120, 116, 115],
    'DDD': 'ABCDEFG'
}, index=[1, 2, 3, 4, 5, 6])

2.loc 标签索引

loc通过标签在DataFrame中选取数据

2.1 loc 获取行

2.1.1 loc 获取一行

print(df)
print("=======================")
# 获取一行数据
print(df.loc[1])

2.1.2 loc 获取多行

print(df)
print("=======================")
print(df.loc[[1, 3]])

2.1.3 loc 获取多行(切片)

print(df)
print("=======================")
print(df.loc[1:5])

2.2 loc获取指定数据(行&列)

当对行和列同时指定时，如果指定值不连续，则需要放在一个列表中；如果指定值是连续的，并采用切片的方式，则不需要加方括号。loc的参数中，左边表示行，右边表示列。

示例一

print(df)
print("=======================")
print(df.loc[2:4, ['AAA', 'CCC']])

示例二

print(df)
print("=======================")
print(df.loc[[1, 3], ['BBB', 'DDD']])

示例三

print(df)
print("=======================")
print(df.loc[:, 'BBB':])

3. iloc 位置索引

loc通过位置在DataFrame中选取数据

3.1 iloc 获取行

3.1.1 iloc 获取单行

以获取第二行为例

print(df)
print("=======================")
print(df.iloc[1])  # 第2行

3.1.2 iloc 获取多行

获取下标为0,2的行（第1、3行）

print(df)
print("=======================")
print(df.iloc[[0, 2]])  # 第1、3行。

获取下标为1到3的行（第2、3、4行）

print(df)
print("=======================")
print(df.iloc[1: 4])  # 第2、3、4行。

获取下标为1的行,及其后边的所有行

print(df)
print("=======================")
print(df.iloc[1:])  # 第二行及以后。

3.2 iloc获取指定数据(行&列) 获取所有行，指定列

print("=======================")
print(df.iloc[:, [1, 3]])

获取所有行，指定连续的列

print("=======================")
print(df.iloc[:, :2])

获取指定行，指定列

print("=======================")
print(df.iloc[[2, 5], [1, 3]])

频道导航