ヒートマップ表示

gistfile1.txt

import matplotlib.pyplot as plt
import seaborn as sns

# Set the size of the heatmap
plt.figure(figsize=(20, 15))

# Adjust the heatmap color mapping to use red for high values and white for low values
plt.figure(figsize=(20, 15))
sns.heatmap(transition_probability_matrix.iloc[:, 1:], cmap='Reds')

plt.title('Transition Probability Matrix Heatmap')
plt.xlabel('Destination Station')
plt.ylabel('Start Station')

# Display the heatmap
plt.show()