用python写一段程序,当data的行数不为空时,查找每行的列,如果元素相同,提取并形成新的dataframe
```python
import pandas as pd
data = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
if len(data) != 0:
df = pd.DataFrame(data) # 将数据转换成DataFrame格式
new_df = df[df.iloc[:,0] == df.iloc[:,1]] # 查找每行的列,如果元素相同,提取并形成新的dataframe
print(new_df)
```
免责声明:
以上内容除特别注明外均来源于网友提问,创作工场回答,未经许可,严谨转载。