初學python，寫下自己寫的圓周率小數(shù)點后1024位熱圖，還請大家批評指正

from mpmath import mp

#取pi的1024位小數(shù)

mp.dps = 1025

pi = mp.pi

print(pi)

# 轉換為字符串

pi = str(pi)

# 去除小數(shù)點

pi = pi.replace('.', '')

# 取后1024位

pi_1024 = pi[1:]

print(pi_1024)

#把整數(shù)分割為數(shù)組

pi_1024L = list(pi_1024)

print(pi_1024L)

#字符串數(shù)組轉換為數(shù)字數(shù)組

pi_1024N = [ int(x) for x in pi_1024L ]

print(pi_1024N)

#一維數(shù)組轉多維數(shù)組

array = np.array(pi_1024N)

pi_array = array.reshape(32,32)

print(pi_array)

import seaborn as sns

import matplotlib.pyplot as plt

sns.set(rc = {'figure.figsize':(6,5)})

sns.heatmap(pi_array, cmap='RdYlBu_r', xticklabels = False, yticklabels = False)

plt.show()