グラフ

Tweet

ヒストグラムを書く方法

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

def main():
    # CSVファイルを読み込む
    df = pd.read_csv("./data/train_data.csv")

    # 'cover'列のヒストグラムを描画
    sns.histplot(df['cover'].values)

    # グラフを表示
    plt.show()

if __name__ == '__main__':
    main()

説明:
- df['cover'].valuesでcover列のデータを取得し、seabornのhistplotでヒストグラムを作成しています。
- plt.show()でグラフを表示します。

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

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

def main():
    # CSVファイルを読み込む
    df = pd.read_csv("./data/train_data.csv")

    # 'year'列で重複を削除し、年ごとのデータを取得
    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]

        # 'cover'列のヒストグラムを描画し、x軸の範囲を指定
        sns.histplot(dy['cover'].values)
        plt.xlim(0, 1.0)  # x軸の範囲を0から1に設定

        # ヒストグラムを保存
        plt.savefig(f"./fig_hist/{year}.png")

        # 現在の図をクリア
        plt.clf()

if __name__ == '__main__':
    main()

説明:
- plt.xlim(0, 1.0)でx軸の範囲を0から1に固定しています。
- 各年ごとのデータに対してヒストグラムを作成し、plt.savefig()で保存しています。

グラフを保存する方法

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

def main():
    # CSVファイルを読み込む
    df = pd.read_csv("./data/train_data.csv")

    # 'year'列で重複を削除し、年ごとのデータを取得
    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]

        # 'cover'列のヒストグラムを描画
        sns.histplot(dy['cover'].values)

        # ヒストグラムを保存
        plt.savefig(f"./fig_hist/{year}.png")

        # 現在の図をクリア
        plt.clf()

if __name__ == '__main__':
    main()

説明:
- 各年ごとにヒストグラムを作成し、plt.savefig()で指定のフォルダに保存しています。
- plt.clf()を使用して、次のループに入る前にグラフをクリアしています。