機器中的齒輪：數(shù)據(jù)包含當前 12 個月的數(shù)據(jù)并水平堆疊。每個月都會修改更新并附加新的月份。 ID |Date |Month1_a |Month1_b |Month1_c |Month2_a |Month2_b |Month2_c |Month3_a |Month3_b |Month3_c ## |MM/DD/YYYY |abc |zxy |123 |NULL |zxy |122 |abc |zxy |123實際數(shù)據(jù)文件沒有標題，并作為每月第 1 個月的不同文件等在下游攝取。ID | Date |Month1_a |Month1_b |Month1_c |New Column## |MM/DD/YYYY |abc |zxy |123 | #ID | Date |Month2_a |Month2_b |Month2_c |New Column## |MM/DD/YYYY |NULL |zxy |122 | #除了復(fù)制文件 12 次。是否有任何關(guān)于閱讀一次并循環(huán)創(chuàng)建我的輸出的建議。我已經(jīng)制定了第 1 個月的邏輯，我對如何進入第 2 個月以上感到困惑。最初是想讀取文件 > 刪除 Month 3+ > Drop Month 1 > Run Logic，但我不確定是否有更好/最佳實踐。謝謝。
1 回答
寶慕林4294392
TA貢獻2021條經(jīng)驗 獲得超8個贊
這將輸出 n 個 csv 文件，其中 n 是輸入數(shù)據(jù)中的月數(shù)。希望這就是您所追求的。
import pandas as pd
df = pd.read_csv('my_data.csv', sep='|')
# Strip whitespace from column names
df.columns = [x.strip() for x in df.columns]
# Get a set of months in the data by splitting on _ and removing 'Month' from
# the first part
months = set([x.split('_')[0].replace('Month','') for x in df.columns if 'Month' in x])
# For each numeric month in months, add those columns with that number in it to
# the ID and Date columns and write to a csv with that month number in the csv title
for month in months:
base_columns = ['ID','Date']
base_columns.extend([x for x in df.columns if 'Month'+month in x])
df[base_columns].to_csv(f'Month_{month}.csv', index=False)
添加回答
舉報
0/150
提交
取消