ทรีแมป

import numpy as np
import matplotlib.pyplot as plt
import japanize_matplotlib
import squarify

np.random.seed(0)  # Fix random numbers

labels = ["A" * i for i in range(1, 5)]
sizes = [i * 10 for i in range(1, 5)]
colors = ["#%02x%02x%02x" % (i * 50, 0, 0) for i in range(1, 5)]

plt.figure(figsize=(5, 5))
squarify.plot(sizes, color=colors, label=labels)
plt.axis("off")
plt.show()

import pandas as pd

data = [
    ["PBR", 80.20, 130.00],
    ["GOOG", 1188.0, 1588.0],
    ["FLNG", 70.90, 230.00],
    ["ZIM", 400.22, 630.10],
    ["GOGL", 120.20, 90.90],
    ["3466\n??????", 156.20, 147.00],  # ??????????
]

df = pd.DataFrame(data)
df.columns = ["驫俶氛蜷・, "蜿門ｾ嶺ｾ｡鬘・, "迴ｾ蝨ｨ縺ｮ萓｡鬘・]
df["隧穂ｾ｡謳咲寢"] = df["迴ｾ蝨ｨ縺ｮ萓｡鬘・] - df["蜿門ｾ嶺ｾ｡鬘・]
df.head(6)

plt.show()

colors = []
percents = []
for p_or_l, oac in zip(df["????"], df["????"]):
    percent = p_or_l / oac * 100

    if p_or_l > 0:
        g = np.min([percent * 255 / 100 + 100, 255.0])
        color = "#%02x%02x%02x" % (0, int(g), 0)
        colors.append(color)
    else:
        r = np.min([-percent * 255 / 100 + 100, 255])
        color = "#%02x%02x%02x" % (int(r), 0, 0)
        colors.append(color)

    percents.append(percent)

print(df["驫俶氛蜷・"].values)
print(colors)
print(percents)

plt.show()

current_prices = [cp for cp in df["迴ｾ蝨ｨ縺ｮ萓｡鬘・]]
labels = [
    f"{name}\n{np.round(percent, 2)}・・".replace("-", "笆ｼ")
    for name, percent in zip(df["驫俶氛蜷・"], percents)
]

plt.figure(figsize=(10, 10))
plt.rcParams["font.size"] = 18
squarify.plot(current_prices, color=colors, label=labels)
plt.axis("off")
plt.show()

plt.figure(figsize=(10, 10))
plt.rcParams["font.size"] = 18
squarify.plot(
    current_prices + [3500], color=colors + ["#ccc"], label=labels + ["?????"]
)
plt.axis("off")
plt.show()