Computer >> บทช่วยสอนคอมพิวเตอร์ >  >> การเขียนโปรแกรม >> Python

วิธีเปลี่ยนเส้นทางเริ่มต้นสำหรับการบันทึกรูปใน Matplotlib

การเปลี่ยนไดเรกทอรีเริ่มต้นสำหรับ savefig

ใช้ rcParams["savefig.directory"] เพื่อกำหนดโฟลเดอร์เริ่มต้นเมื่อบันทึกรูปภาพ

ขั้นตอนหลัก

  • ตั้งค่าขนาดรูปและ padding
  • สร้างข้อมูลสุ่มด้วย numpy
  • ใช้ imshow() แสดงภาพ
  • บันทึกรูปด้วย plt.savefig()

ตัวอย่างโค้ด

import os
import numpy as np
from matplotlib import pyplot as plt

plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True

dir_name = "C:/Windows/Temp/"
plt.rcParams["savefig.directory"] = os.chdir(os.path.dirname(dir_name))

data = np.random.rand(5, 5)
plt.imshow(data, cmap="copper")
plt.savefig("img.png")

ผลลัพธ์

ไฟล์ "img.png" จะถูกบันทึกที่เส้นทาง "C:/Windows/Temp/"

หมายเหตุ

os.chdir() จะเปลี่ยน working directory ของ process ปัจจุบัน