Lưu hình matplotlib/networkx không có lề

Question

Khi tôi vẽ một hình bằng cách sử dụng matplotlib làm cách nào để lưu hình mà không có thêm lề? Thông thường khi tôi lưu nó như

plt.savefig("figure.png") # or .pdf 

tôi nhận được nó với một số lề:

Ví dụ:

import matplotlib.pyplot as plt 
import networkx as nx 

G=nx.Graph() 

G.add_edge('a','b',weight=1) 
G.add_edge('a','c',weight=1) 
G.add_edge('a','d',weight=1) 
G.add_edge('a','e',weight=1) 
G.add_edge('a','f',weight=1) 
G.add_edge('a','g',weight=1) 

pos=nx.spring_layout(G) 
nx.draw_networkx_nodes(G,pos,node_size=1200,node_shape='o',node_color='0.75') 

nx.draw_networkx_edges(G,pos, 
       width=2,edge_color='b') 

plt.axis('off') 
plt.savefig("degree.png", bbox_inches="tight") 
plt.show() 

Cập nhật 2:

Các khoảng trống được đặt bên trong trục .. Điều này rõ ràng nếu tôi xóa plt.axis('off')
Vì vậy, tôi nghĩ có một số mẹo để sử dụng với gói Networkx.

Answer 1

thêm các mã bên dưới để kiểm soát giới hạn lô trước tiết kiệm.

thử các giá trị khác nhau của cut, như từ 1,05 đến 1,50, cho đến khi bạn thấy phù hợp.

# adjust the plot limits 
cut = 1.05 
xmax= cut*max(xx for xx,yy in pos.values()) 
ymax= cut*max(yy for xx,yy in pos.values()) 
plt.xlim(0,xmax) 
plt.ylim(0,ymax) 

Answer 2

Hãy thử plt.savefig("figure.png", bbox_inches="tight").

Chỉnh sửa: Ah, bạn đã không đề cập đến bạn đang sử dụng networkx (mặc dù bây giờ tôi thấy nó được liệt kê trong thẻ). bbox_inches="tight" là cách cắt hình chặt chẽ. Tôi không biết những gì networkx đang làm, nhưng tôi tưởng tượng nó thiết lập một số tham số cốt truyện được thêm không gian thêm cho các trục. Bạn nên tìm một giải pháp trong networkx hơn là matplotlib. (Ví dụ: mạngx có thể thêm không gian bên trong các trục , không phải hình, nó trông như thế nào nếu bạn xóa cuộc gọi axis('off')?)

Answer 3

Mà không biết các chi tiết cụ thể của networkx tôi không thể chắc chắn rằng điều này sẽ làm việc, nhưng để loại bỏ các khoảng trắng hoàn toàn từ bên ngoài của một trục trong matplotlib, bạn có thể làm điều gì đó dọc theo dòng:

import matplotlib.pyplot as plt 
ax = plt.axes([0, 0, 1, 1]) 
plt.plot(range(10)) 

Answer 4

Một chút hack, nhưng nếu bạn sử dụng nx.draw, thì nó sẽ phù hợp hơn.

Vì vậy, bạn có thể làm

nx.draw(G,pos,node_size=1200,node_shape='o',node_color='0.75', edgelist = []) 

mà chỉ thu hút các nút và không có các cạnh. Sau đó, bạn có thể thực hiện

nx.draw_networkx_edges(G, pos, width=2, edge_color='b')