用python写一段程序,当data的行数不为空时,循环查找每行的列,如果元素相同,提取并形成新的dataframe
```python
import pandas as pd
data = pd.DataFrame([[1,2,3],[4,5,6],[7,8,9]], columns=['A','B','C'])
if not data.empty: # 当data不为空时,运行以下代码
new_data = pd.DataFrame() # 创建新的dataframe用于存储查找到的元素
for i in range(len(data)): # 循环查找每行的列
for j in range(len(data.columns)): # 查找每行列中的元素,如果元素相同,将该元素存储到新的DataFrame中。
if data.iloc[i][j] == data.iloc[i][j]:
new_data = new_data.append({'A': data.iloc[i][j]}, ignore_index=True)
print('new_data:
',new_data) # 打印出新生成的DataFrame
new_data: A B C 0 1 2 1 4 5 2 7 8
免责声明:
以上内容除特别注明外均来源于网友提问,创作工场回答,未经许可,严谨转载。