Pandas数据切片

以下是Pandas中常见的数据切片方法:

1. 使用切片操作符(:):

可以使用切片操作符对行和列进行切片,例如选择前5行和前2列:

import pandas as pd

# 创建一个DataFrame

data = {‘A’: [1, 2, 3, 4, 5, 6],

‘B’: [10, 20, 30, 40, 50, 60],

‘C’: [100, 200, 300, 400, 500, 600]}

df = pd.DataFrame(data)

# 使用切片操作符切片前5行和前2列

result = df[:5][[‘A’, ‘B’]]

print(result)

# 输出

#    A   B

# 0  1  10

# 1  2  20

# 2  3  30

# 3  4  40

# 4  5  50

2. 使用iloc函数:

iloc函数是基于整数位置索引的切片方法,可以使用行和列的整数位置索引进行切片。例如,选择前3行和前1列:

import pandas as pd

# 创建一个DataFrame

data = {‘A’: [1, 2, 3, 4, 5, 6],

‘B’: [10, 20, 30, 40, 50, 60]}

df = pd.DataFrame(data)

# 使用iloc函数切片前3行和前1列

result = df.iloc[:3, :1]

print(result)

# 输出

#    A

# 0  1

# 1  2

# 2  3

3. 使用loc函数:loc函数是基于标签名称索引的切片方法,可以使用行和列的标签名称进行切片。例如,选择前3行和前1列:

import pandas as pd

# 创建一个DataFrame

data = {‘A’: [1, 2, 3, 4, 5, 6],

‘B’: [10, 20, 30, 40, 50, 60]}

df = pd.DataFrame(data)

# 使用loc函数切片前3行和前1列

result = df.loc[:2, [‘A’]]

print(result)

# 输出

#    A

# 0  1

# 1  2

# 2  3

4. 使用at和iat函数:

at和iat函数是基于标签名称和整数位置索引的切片方法,可以分别用于选择单个元素。例如,选择第一行第一列的元素:

import pandas as pd

# 创建一个DataFrame

data = {‘A’: [1, 2, 3, 4, 5, 6],

‘B’: [10, 20, 30, 40, 50, 60]}

df = pd.DataFrame(data)

# 使用at函数选择第一行第一列的元素

result = df.at[0, ‘A’]

print(result)

# 输出

# 1

# 使用iat函数选择第一行第一列的元素

result = df.iat[0, 0]

print(result)

# 输出

# 1

以上是Pandas中常见的数据切片方法,可以根据实际需求选择合适的方法。

© 版权声明
THE END
喜欢就支持一下吧
点赞13 分享
评论 抢沙发
头像
欢迎您留下宝贵的见解!
提交
头像

昵称

取消
昵称表情代码图片