การย้ายถิ่นระหว่างรัฐ
ในวิดีโอ คุณได้เห็น heatmap ของการย้ายถิ่นระหว่างรัฐ แต่ข้อมูลยังดูรกอยู่ ในแบบฝึกหัดนี้ จะเจาะดูเฉพาะการเคลื่อนย้าย ภายใน ภาคมิดเวสต์เท่านั้น
ได้โหลด DataFrame ชื่อ state_to_state ไว้แล้ว และแสดงแถวแรกบางส่วนในคอนโซล จำไว้จากวิดีโอว่า ป้ายชื่อแถวแสดงรัฐที่ ย้ายไป ส่วนชื่อคอลัมน์แสดงรัฐที่ ย้ายมาจาก
มีการกำหนดลิสต์ midwest_states ที่ประกอบด้วยชื่อรัฐในภาคมิดเวสต์ไว้แล้ว (สามารถ print เพื่อดูรายชื่อรัฐได้) DataFrame นี้ใช้ชื่อรัฐเป็นทั้งชื่อคอลัมน์และ index ดังนั้นจะใช้ midwest_states เพื่อดึงคอลัมน์และแถวที่ต้องการสำหรับ heatmap นี้
นำเข้า pandas และ seaborn ด้วย alias ตามปกติแล้ว
แบบฝึกหัดนี้เป็นส่วนหนึ่งของหลักสูตร
การวิเคราะห์ข้อมูลสำมะโนประชากรสหรัฐฯ ด้วย Python
คำแนะนำการฝึกหัด
- Subset DataFrame เพื่อดึงเฉพาะคอลัมน์ที่ตรงกับรายชื่อรัฐในลิสต์ และเฉพาะแถวที่มี index อยู่ในลิสต์รัฐนั้น
- การ subset อาจทำให้ลำดับคอลัมน์และแถวเปลี่ยนไป ตรวจสอบว่า
midwest.indexเท่ากับmidwest.columnsหรือไม่ - เรียงลำดับ DataFrame ตาม row index (
axis = 0) และตามชื่อคอลัมน์ (axis = 1) โดยใช้inplace = Trueในทั้งสองกรณี - สร้าง heatmap ของ
midwestโดยใช้ชุดสีเหลือง-เขียว-น้ำเงิน ด้วยcmap="YlGnBu"
แบบฝึกหัดเชิงโต้ตอบแบบลงมือทำ
ลองทำแบบฝึกหัดนี้โดยเติมโค้ดตัวอย่างนี้ให้สมบูรณ์
# Retain only rows and columns of Midwest states
midwest = state_to_state[____][state_to_state.index.isin(____)]
# Are rows and columns still in the same order?
print(____)
# Sort the rows (by index) and columns (by name)
midwest.sort_index(axis = ____, ____)
midwest.sort_index(axis = ____, ____)
# Create a heatmap of migration flows
____
plt.xticks(rotation=90)
plt.yticks(rotation=0)
plt.show()