Я хочу сохранить таблицу DataFrame как изображение, и меня вдохновляет этот вопрос .
Но, к сожалению, нарисованное мной изображение таблицы настолько расплывчато, и я не могу его прояснить.
Это моя среда:
Python 3.7.0
pandas 0.23.0
matplotlib 2.2.2
А вот мой код, его таблица и моя фотография:
train_corr
Survived Pclass Age SibSp Parch Fare
Survive 1.000000 -0.338481 -0.077221 -0.035322 0.081629 0.257307
Pclass -0.338481 1.000000 -0.369226 0.083081 0.018443 -0.549500
Age -0.077221 -0.369226 1.000000 -0.308247 -0.189119 0.096067
SibSp -0.035322 0.083081 -0.308247 1.000000 0.414838 0.159651
Parch 0.081629 0.018443 -0.189119 0.414838 1.000000 0.216225
Fare 0.257307 -0.549500 0.096067 0.159651 0.216225 1.000000
from pandas.plotting import table
ax = plt.subplot(111, frame_on=False) # no visible frame
ax.xaxis.set_visible(False) # hide the x axis
ax.yaxis.set_visible(False) # hide the y axis
table(ax, train_corr, loc='center')
plt.savefig('correlation_matrix_HD.jpg')

И я искал документацию о пандах.plotting и matplotlib.pyplot.table , но я до сих пор не знаю, как это выяснить.
Если не возражаете, кто-нибудь может мне помочь?
Искренне спасибо.