如何在python中为复制pandas数据帧编写for循环

Question

我是数据工程师的新手。现在，我尝试为pandas数据帧中的重复数据编写python代码。例如数据：

A B C D E F G E
1 2 3 4 0 1 0 1
5 6 7 8 0 1 1 0
9 1 2 3 0 1 0 1

我需要将数据帧复制到

dfE =  A B C D E 
1 2 3 4 0 
5 6 7 8 0 
9 1 2 3 0
dfF =  A B C D F 
1 2 3 4 1 
5 6 7 8 1 
9 1 2 3 1
dfG...

请帮帮我。。。

Answer 1

你好，欢迎来到的stackoverflow

基本上panda允许选择具有列名的列，您可以查看下面的代码，这可能会解决您的问题。

from io import StringIO
import pandas as pd
data = """
A B C D E F G E
1 2 3 4 0 1 0 1
5 6 7 8 0 1 1 0
9 1 2 3 0 1 0 1
"""
df = pd.read_csv(StringIO(data), sep=' ')
dfE = df[['A', 'B', 'C', 'D', 'E']]
dfF = df[['A', 'B', 'C', 'D', 'F']]
dfG = df[['A', 'B', 'C', 'D', 'G']]