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

เขียนโปรแกรม Python ส่งออก DataFrame เป็นไฟล์ Pickle และอ่านคืนค่า

สมมติว่าคุณมี DataFrame และต้องการบันทึกเป็นไฟล์ Pickle แล้วอ่านคืนมาใช้งาน

วิธีแก้ปัญหา

  1. สร้าง DataFrame
  2. บันทึกเป็น pickle ด้วย df.to_pickle('pandas.pickle')
  3. อ่านคืนด้วย pd.read_pickle('pandas.pickle')

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

import pandas as pd

df = pd.DataFrame({'Fruits': ["Apple", "Orange", "Mango", "Kiwi"],
                   'City': ["Shimla", "Sydney", "Lucknow", "Wellington"]})

print("ส่งออกไปยังไฟล์ pickle...")
df.to_pickle('pandas.pickle')

print("อ่านคืนจากไฟล์ pickle:")
result = pd.read_pickle('pandas.pickle')
print(result)

ผลลัพธ์

ส่งออกไปยังไฟล์ pickle...
อ่านคืนจากไฟล์ pickle:
   Fruits      City
0   Apple    Shimla
1  Orange   Sydney
2   Mango  Lucknow
3   Kiwi  Wellington

คำอธิบาย:

  • Pickle format: binary format ของ Python สำหรับ serialize object ได้ทุกประเภท รวมถึง DataFrame ที่มี dtype ซับซ้อน
  • ข้อดี: รักษา dtype, index, column names ได้สมบูรณ์ เร็วกว่า CSV
  • ข้อเสีย: ไม่ใช่ human-readable, ใช้ได้เฉพาะ Python, มีความเสี่ยงด้านความปลอดภัยถ้าโหลดไฟล์จากแหล่งไม่ไว้ใจ
  • เหมาะสำหรับ: การเก็บข้อมูลชั่วคราว (cache), ส่งผ่านระหว่าง process ใน Python