グラフ

Tweet

ヒストグラムを書く方法

# coding: UTF-8
import pandas as pd
import matplotlib.pyplot as plt
import seaborn as sns


def main():
    df = pd.read_csv("./data/train_data.csv")
    sns.histplot(df['cover'].values)
    plt.show()


if __name__ == '__main__':
    main()

ヒストグラムでx軸の範囲を指定する方法

# coding: UTF-8
import pandas as pd
import matplotlib.pyplot as plt
import seaborn as sns


def main():
    df = pd.read_csv("./data/train_data.csv")
    df_year = df.drop_duplicates(subset='year')
    print(df_year['year'].values)
    for year in df_year['year'].values:
        print(year)
        dy = df[df['year'] == year]
        sns.histplot(dy['cover'].values)
        plt.xlim(0, 1.0)
        plt.savefig(f"./fig_hist/{year}.png")
        plt.clf()


if __name__ == '__main__':
    main()

グラフを保存する方法

# coding: UTF-8
import pandas as pd
import matplotlib.pyplot as plt
import seaborn as sns


def main():
    df = pd.read_csv("./data/train_data.csv")
    df_year = df.drop_duplicates(subset='year')
    print(df_year['year'].values)
    for year in df_year['year'].values:
        print(year)
        dy = df[df['year'] == year]
        sns.histplot(dy['cover'].values)
        plt.savefig(f"./fig_hist/{year}.png")
        plt.clf()


if __name__ == '__main__':
    main()