Python pandas 对DataFrame进行遍历(持续更新)

夫琅禾费米线 2024-08-17 17:35:07 阅读 88

三种用迭代器的方法(items,iterrows,itertuples):沿着行/列对所有元素遍历,适合对每个元素进行操作简单dolumns,loc遍历:遍历每一行/列,适合对每行/列进行求和、离散等整体操作for+zip: 适合选取较少的特定行/列进行操作

       之后看到其他好用方法,或者合适的拓展也会继续更新

        有错误请指正,欢迎评论建议

items,iterrows,itertuples:

items(): 以 <code>(列名, Series) 对的形式遍历 DataFrame 的列。

函数原型:DataFrame.iteritems(None)

返回:返回一个迭代器,产生 (列名,列内容Series) 对的元组。

        

import pandas as pd

data = {'A': [1, 2, 3], 'B': [4, 5, 6]}

df = pd.DataFrame(data)

for column, series in df.items():

print(f"Column: {column}")

print(f"Series:\n{series}\n")

输出:

iterrows():以 <code>(行下标, Series) 对的形式遍历 DataFrame 的行。

        原型:DataFrame.iterrows()

        返回:返回迭代器(行号index,行内数据Series)

import pandas as pd

data = {'A': [1, 2,], 'B': [4, 5,]}

df = pd.DataFrame(data)

for index, row in df.iterrows():

# 处理每一行的逻辑

print(f"row_index: {index}")

print(f"eow_items\n:{row}\n")

输出:

 

itertuples():以命名元组的方式遍历行

        函数原型:DataFrame.itertuples(index=Truename='Pandas')

                        index: True则返回的tuple中首个元素为行号

                        name: 字符串或者None,作为返回的tuple的名字,如果为None则返回常规tuple

        返回:一个迭代器,每一行的命名元组(name,row_values)

<code>import pandas as pd

data = {'A': [1, 2,], 'B': [4, 5,]}

df = pd.DataFrame(data)

###1.默认

for row in df.itertuple():

print(row)

'''输出

Pandas(Index=0, A=1, B=4)

Pandas(Index=1, A=2, B=5)

'''

###

##2.不输出index

for row in df.itertuples(index=False):

print(row)

'''输出

Pandas(A=1, B=4)

Pandas(A=2, B=5)

'''

###

##3. 输出常规tuple

for row in df.itertuples(name=None):

print(row)

'''输出

(0, 1, 4)

(1, 2, 5)

'''

 columns+values,index

columns:获取dataframe的列标签

        返回:<class 'pandas.core.indexes.base.Index'> (不重要,理解为列标签,获取到列名和数据类型,但无法直接进行操作)

         如果想要获取列标签的数组,可以用df.columns.values,想得到list类型就加上tolist()

import pandas as pd

data = {'A': [1, 2,], 'B': [4, 5,],'C':[6,7]}

df = pd.DataFrame(data)

print('column names and type: ',df.columns)

'''

输出:

column names and type: Index(['A', 'B', 'C'], dtype='object')code>

'''

print('column name: ',df.columns.values)#数组类型。

'''

输出:

column name: ['A' 'B' 'C']

'''

print('column name: ',df.columns.tolist())#等同于list(df.columns)

print('column name: ',list(df))

print('column name: ',list(df.columns))

'''

以上三种均输出

column name: ['A', 'B', 'C']

'''

index:DataFrame的索引标签

        和columns类似,在这里就不赘述了。

        对每行/每列进行某操作时,常用这两个方法。但是很多时候可以被df.apply(function,axis=0)替代(axis默认为0,0沿着列操作,1沿着行操作)

import pandas as pd

data = {'A': [1, 2,], 'B': [4, 5,],'C':[6,7]}

df = pd.DataFrame(data)

# 使用 columns 遍历 DataFrame 的列名

print("columns:")

for column in df.columns:

print(column)

'''输出

columns:

A

B

C

'''

# 使用 index 遍历 DataFrame 的行索引

print("\nindexs:")

for index in df.index:

print(index)

'''输出

indexs:

0

1

'''

for+zip:取特定几列/行遍历

这个方法很简单,运行效率高。但是只适用于对特定几行/列遍历,如果列或行较多的时候书写很繁琐,且可能因个人粗心原因出错。

import pandas as pd

data = {'A': [1, 2,], 'B': [4, 5,],'C':[6,7]}

df = pd.DataFrame(data)

#对特定几列遍历

#for a,b,c in zip([df[col] for col in df.columns])

for a,b in zip(df['A'],df['B']):

print('a: ',a,' b:',b)

'''输出

a: 1 b: 4

a: 2 b: 5

'''

#对特定几行遍历

for row1,row2 in zip(df.iloc[0],df.iloc[1]):#如果时标签索引则用df.loc['RowName']

print('row1: ',row1,' row2:' ,row2)

'''输出

row1: 1 row2: 2

row1: 4 row2: 5

row1: 6 row2: 7

'''



声明

本文内容仅代表作者观点,或转载于其他网站,本站不以此文作为商业用途
如有涉及侵权,请联系本站进行删除
转载本站原创文章,请注明来源及作者。