簡體   English   中英

使用matplotlib在python代碼for-loop中將折線圖轉換為條形圖

[英]Turn line chart into bar plot in python code for-loop with matplotlib

我有一個腳本,需要多個.csv文件並輸出多個折線圖。 我希望這些圖成為條形圖,因為它們顯示降雨。 但是我無法使plot.bar.()工作。 到目前為止,這是對我有用的代碼,但這是不正確的。 有什么幫助嗎?

import pandas as pd
import time
import os
import matplotlib.pyplot as plt

files = ['w.pod.csv',
    't.pod.csv',
    'r.pod.csv',
    'n.pod.csv',
    'm.pod.csv',
    'k.pod.csv',
    'j.pod.csv',
    'h.pod.csv',
    'g.pod.csv',
    'c.pod.csv',
    'b.pod.csv']

for f in files:
    fn = f.split('.')[0]
    dat = pd.read_csv(f)
    df0 = dat.loc[:, ['TimeStamp', 'RF']]
    # Change time format
    df0["time"] = pd.to_datetime(df0["TimeStamp"])
    df0["day"] = df0['time'].map(lambda x: x.day)
    df0["month"] = df0['time'].map(lambda x: x.month)
    df0["year"] = df0['time'].map(lambda x: x.year)
    df0.to_csv('{}_1.csv'.format(fn), na_rep="0")  # write to csv

    # Combine for daily rainfall
    df1 = pd.read_csv('{}_1.csv'.format(fn), encoding='latin-1',
                  usecols=['day', 'month', 'year', 'RF', 'TimeStamp'])
    df2 = df1.groupby(['day', 'month', 'year'], as_index=False).sum()
    df2.to_csv('{}_2.csv'.format(fn), na_rep="0", header=None)  # write to csv

    # parse date
    df3 = pd.read_csv('{}_2.csv'.format(fn), header=None, index_col='datetime',
                 parse_dates={'datetime': [1,2,3]},
                 date_parser=lambda x: pd.datetime.strptime(x, '%d %m %Y'))

    def dt_parse(date_string):
        dt = pd.datetime.strptime(date_string, '%d %m %Y')
        return dt

    # sort datetime
    df4 = df3.sort()
    final = df4.reset_index()

    # rename columns
    final.columns = ['date', 'bleh', 'rf']

    final[['date','rf']].plot()
    plt.suptitle('{} Rainfall 2015-2016'.format(fn), fontsize=20)
    plt.xlabel('Date', fontsize=18)
    plt.ylabel('Rain / mm', fontsize=16)
    plt.savefig('{}.png'.format(fn))

在此處輸入圖片說明

這是我先前問題的擴展: 使用多個.csv文件在python中自動制作多個圖

使用DataFrame.plot.bar

final[['date','rf']].plot.bar()

DataFrame.plot

final[['date','rf']].plot(kind='bar')

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM