如何在 python 中編寫循環以復制 pandas dataframe

Question

我是數據工程師的新手。 現在我嘗試為來自 pandas dataframe 的重復數據編寫 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

我需要將 dataframe 復制到

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

嗨 piyaphong 歡迎來到 stackoverflow，

基本上 pandas 允許 select 列具有列名，您可以查看下面的代碼，這可能會解決您的情況。

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']]