График данных хранится в словаре в цикле - PullRequest
0 голосов
/ 10 марта 2020

Я загрузил 5 данных в список из csvs. Чтобы убедиться, что каждый фрейм данных был помечен правильно, я создал словарь с ключом в качестве имени df (из Gage_lst) и значением в качестве каждого df (из csv). Теперь я хочу построить данные временных рядов из 5 кадров данных, хранящихся в словаре. Я хотел бы использовать al oop для построения графика зависимости времени от времени для каждого кадра данных и пометить график правильной меткой из соответствующего ключа словаря.

Gage_lst= ['COR.txt','DWN.txt','ORR.txt','ARK.txt','HK.txt','UP.txt','PCR.txt','AKRK.txt']

dfs=[]
for gage in Gage_lst:
    gagefile = os.path.join(gage_dir,gage)
    df = pd.read_csv(gagefile, skiprows=1, header=None, delim_whitespace=True, names=['time', 'stage', 'flow'])
    dfs.append(df)

for df in dfs:    
    timestep = []
    for i in range(0,len(df)):
        timestep.append(i)
    df['StressPeriod'] = timestep 
    df.set_index('StressPeriod',inplace=True)

    gage_names = dict(zip(Gage_lst, dfs)) 

for key, value in gage_names.items():
    fig,ax = plt.subplots()
    df.flow.plot(legend=True,title = 'NONE', figsize=(20, 6))
    plt.ylabel('flow')
    plt.xlabel('StressPeriod')
    plt.title(key)
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...