【小ネタ】uniqコマンドによる集計(カウント)結果を入力にグラフを描画するPythonスクリプト

import sys
import matplotlib.pyplot as plt

def read_data():
    """
    標準入力からデータを読み込み、(項目, 数値)のリストとして返す
    """
    data = []
    for line in sys.stdin:
        count, item = line.strip().split(None, 1)
        data.append((item, int(count)))
    return data

def create_horizontal_bar_chart(data):
    """
    データを入力に棒グラフ(横向き)を作成し表示する
    """
    items, counts = zip(*data)
    fig, ax = plt.subplots()

    ax.barh(items, counts) # 横向きの棒グラフを作成
    ax.invert_yaxis() # 上から順番に表示させる
    plt.tight_layout() # 余白調整
    plt.show()

if __name__ == "__main__":
    data = read_data()
    create_horizontal_bar_chart(data)

$ sort fruits.txt | uniq -c | sort -nr
   8 apple
   6 orange
   5 banana
   2 strawberry
   2 grape

$ sort fruits.txt | uniq -c | sort -nr | python3 bar_chart.py
# ↓以下画像↓

aws cloudwatch list-metrics --query "Metrics[].[Namespace]" --output text \
| sort | uniq -c | sort -nr
# 434 AWS/Usage
# 349 AWS/Config
#  42 AWS/Lambda
#  18 AWS/Events
#  16 AWS/S3
#  12 AWS/DynamoDB
#  11 AWS/Logs
#   8 AWS/States
#   8 AWS/SNS
#   8 AWS/Location
#   6 AWS/Rekognition
#   3 AWS/Glue
#   2 AWS/Scheduler
#   2 AWS/Firehose

aws cloudwatch list-metrics --query "Metrics[].[Namespace]" --output text \
| sort | uniq -c | sort -nr \
| python3 bar_chart.py
# ↓以下画像↓

import sys
import numpy as np
import matplotlib.pyplot as plt
import matplotlib.colors as mcolors

def read_data():
    data = []
    for line in sys.stdin:
        count, item = line.strip().split(None, 1)
        data.append((item, int(count)))
    return data

def create_horizontal_bar_chart(data):
    items, counts = zip(*data)

    # カラーマップを作成（項目数に応じて色を生成）
    colors = plt.cm.Set3(np.linspace(0, 1, len(items)))

    fig, ax = plt.subplots(figsize=(12, 6))
    bars = ax.barh(range(len(items)), counts, color=colors)

    ax.set_xlabel('Count')
    ax.set_title('Item Distribution')
    ax.set_yticks(range(len(items)))
    ax.set_yticklabels(items)

    # バーの右側に数値を表示
    for i, (item, count) in enumerate(data):
        ax.text(count, i, f' {count}', va='center')

    # 凡例を追加
    for bar, item in zip(bars, items):
        bar.set_label(item)
    ax.legend(loc='center left', bbox_to_anchor=(1, 0.5))

    plt.gca().invert_yaxis()  # y軸を反転して、入力順を維持
    plt.tight_layout()
    plt.show()

if __name__ == "__main__":
    data = read_data()
    create_horizontal_bar_chart(data)