numpy: ndarrayを二値画像ではなくグレースケールの画像として保存したい

Question

前提・実現したいこと

numpy.ndarrayに格納している64×64の配列をグレースケールの画像として表示、保存をしたいです.（元々配列を二値化、つまり白黒画像にして保存するものだったがグレースケールに変換するバージョンも作りたい）
matplotlibを用いて表示はできているのですが、上下に透過部分が余分に入ってしまいます。
画像1,2枚の保存だけでなくfor文内で実行して数万枚単位で保存をするため、できれば
①外部のトリミングソフトの使用は控えたい
②手作業で右クリックで保存するのではなくて画像の保存まで自動化したい
と考えています。

発生している問題・エラーメッセージ

これ以降、Aが64×64のnumpy.ndarrayだとします
Aには0～1の範囲の値が格納されているとします

plt.figure(figsize=(5, 5))
ax = plt.gca() # get current axis
ax.spines["right"].set_color("none")  # 右消し
ax.spines["left"].set_color("none")   # 左消し
ax.spines["top"].set_color("none")    # 上消し
ax.spines["bottom"].set_color("none") # 下消し

plt.tick_params(labelbottom=False,
                labelleft=False,
                labelright=False,
                labeltop=False)
plt.tick_params(bottom=False,
                left=False,
                right=False,
                top=False)

plt.imshow(A, cmap='gray')
plt.tight_layout()
plt.savefig('IMAGE.png', transparent=True) 

↑一応画像は保存できるが余分な透過部分が残ってしまう

該当のソースコード

import numpy as np
from matplotlib import pyplot as plt
import cv2
import copy
from scipy import signal
import os
from PIL import Image
import numpy as np
%matplotlib inline
"""
ここのfor文で64×64の配列を生成するに記述されている
"""
A=np.real(B[i,:,:])#Bの配列内の実数部のみをAに移した
type(A)
"""
元々の二値化する方では以下のようにして二値化画像を保存していた。
できればこの部分のみを変えて実現したい
binary=np.real(B[i,:,:])
th=0.0
im_bin=(binary>th)*255
im_bin2=cv2.bitwise_not(im_bin)+256
pilImg = Image.fromarray(np.uint8(im_bin2))
  
pilImg.save("%03.f"%(i)+".png")
"""

試したこと

type(A)

↑numpy.ndarray

A

↑array([[0.82116601, 0.81233283, 0.80962479, ..., 0.81859178, 0.82586342,
0.82116601],
[0.81202854, 0.81069888, 0.80997338, ..., 0.82054398, 0.82182087,
0.81202854],
[0.80295631, 0.8081905 , 0.81753533, ..., 0.81573611, 0.81126605,
0.80295631],
...,
[0.81416119, 0.81354811, 0.81656137, ..., 0.81528227, 0.81099825,
0.81416119],
[0.81679137, 0.81531291, 0.81720691, ..., 0.81377625, 0.81838123,
0.81688567],
[0.82116601, 0.81233283, 0.80962479, ..., 0.81859178, 0.82294009,
0.82116601]])

A.shape

↑(64, 64)

type(A[0,0])

↑numpy.float64

plt.figure(figsize=(5, 5))
ax = plt.gca() # get current axis
ax.spines["right"].set_color("none")  # 右消し
ax.spines["left"].set_color("none")   # 左消し
ax.spines["top"].set_color("none")    # 上消し
ax.spines["bottom"].set_color("none") # 下消し

plt.tick_params(labelbottom=False,
                labelleft=False,
                labelright=False,
                labeltop=False)
plt.tick_params(bottom=False,
                left=False,
                right=False,
                top=False)

plt.imshow(A, cmap='gray')
plt.tight_layout()
plt.savefig('IMAGE.png', transparent=True)

↑画像は保存できるが透過部分が残ってしまう。できれば64×64配列の部分のみを保存したい
また、Aのグレースケールは以下のような画像です(plt.imshow(A, cmap='gray')部分)

Answer 1

float 型の2次元配列を画像として保存したいということでしょうか。
値の範囲を [0, 255] にスケールして、型を uint8 にしてから保存すればよいので、numpy で書いてもいいですが、scikit-image ライブラリを使用するとこの処理がかんたんに行えます。pip でインストールできます。

pip install scikit-image

# pip install scikit-image
import skimage
from skimage import io
import numpy as np

img = np.random.rand(64, 64)
print(img)

# float -> uint8
img = skimage.img_as_ubyte(img)
print(img)

io.imsave("img.png", img)