Pythonで画像データが保存できない

Question

以下のコードで画像データを保存しようとしているのですが、実行時に表示されるデータが画像ファイルとして正しく保存できません（ファイルは作成されますが、開くと中身が空になっています）。
どのようにすれば出力画像を保存することができるでしょうか。

※1　解像度の高い画像を保存することを目的としています
※2　testData.xlsxファイルにはA列とB列にそれぞれアルファベットが入っており、アルファベット間のネットワークが出力されるイメージです

import numpy
import openpyxl
import collections
import random
import matplotlib.pyplot as plt
import networkx as nx
import string
import numpy as np


split = "->"  
maxNodeSize = 600
maxEdgeSize = 4


data = []
nodes = []

wb = openpyxl.load_workbook(filename="testData.xlsx")
ws = wb[wb.sheetnames[0]]
print(ws)

for i in range(len(ws["A"])):
    a = ws["A"+str(i+1)].value
    b = ws["B"+str(i+1)].value
    data.append([a, b])

# adjust data
stringification = []

for i in data:
    nodes.append(i[0])
    nodes.append(i[1])

    if i[0] == i[1]:
        continue
    if i[0] < i[1]:
        stringification.append(i[0]+split+i[1])
    else:
        stringification.append(i[1]+split+i[0])
stringification.sort()

print(nodes)
for i in stringification:
    print(i)


# count data
counterNode = collections.Counter(nodes)
counterEdge = collections.Counter(stringification)

countNodeMax = counterNode.most_common(1)[0][1]
countEdgeMax = counterEdge.most_common(1)[0][1]

print(counterNode)
print(counterEdge)


# # recreate data
nodes = []
nodeSizes = []
for key in counterNode.keys():
    nodes.append(key)
    nodeSizes.append(counterNode[key]*maxNodeSize/countNodeMax)
for i in nodes:
    print(i)

data = []
for key in counterEdge.keys():
    left, right = key.split(split)
    data.append([(left, right), counterEdge[key]*maxEdgeSize/countEdgeMax])
for i in data:
    print(i)

# # create graph
G = nx.Graph()

for node in nodes:
    G.add_node(node)

pos = nx.random_layout(G, seed=numpy.random)
nx.draw_networkx_nodes(G, pos, node_color='r', alpha=1, node_size=nodeSizes)

nx.draw_networkx_labels(G, pos, font_size=10,
                        font_family='MS Gothic')

for data in data:
    nx.draw_networkx_edges(
        G, pos, edgelist=[data[0]],
        width=data[1], edge_color='b', alpha=0.1)
plt.figure(figsize=(8,6),dpi=1000)
plt.savefig("test.png",dpi=600)

Answer 1

再現テストはしていませんが、

plt.figure(figsize=(8,6),dpi=1000)
plt.savefig("test.png",dpi=600)

とやれば、空の画像が保存されるでしょう。

plt.figure(figsize=(8,6),dpi=1000)

をnetworkxで描画する前に移動してみてください。

Comments

[trave]

できました！
ありがとうございます。