サンプルコード: matplotlib.pyplot で出力するグラフのサイズを指定する

render_chart.py

"""サンプルコード: matplotlib.pyplot で出力するグラフのサイズを指定する

次の 2 ステップでできる:

1. `fig.set_size_inches()` で DPI で割ったサイズを指定しておく
2. `pyplot.savefig()` で DPI を指定して出力する
"""

import matplotlib.pyplot as plt
import pandas as pd

CSV_IN = './data.csv'
FILE_OUT = './chart.png'
DPI = 96
OUT_WIDTH = 1000
OUT_HEIGHT = 600


def main():
    df = pd.read_csv(CSV_IN, index_col=0)
    xseries = df.columns
    fig, ax = plt.subplots()
    fig.set_size_inches(OUT_WIDTH / DPI, OUT_HEIGHT / DPI)
    for index, row in df.iterrows():
        line, = ax.plot(xseries, row.values)
        line.set_label(index)
        ax.legend()

    # show_chart()
    save_file()


def show_chart():
    plt.show()


def save_file():
    plt.savefig(FILE_OUT, dpi=DPI)


if __name__ == '__main__':
    main()

サンプル data.csv:

,2019,2020,2021
Dog,5,3,8
Cat,10,20,15
Panda,3,1,3

参考:

• https://stackoverflow.com/questions/13714454/specifying-and-saving-a-figure-with-exact-size-in-pixels