อัปเดตดัชนีหลังจากจัดเรียง data-frame

ใช้กรอบข้อมูลต่อไปนี้:

x = np.tile(np.arange(3),3)
y = np.repeat(np.arange(3),3)
df = pd.DataFrame({"x": x, "y": y})

   x  y
0  0  0
1  1  0
2  2  0
3  0  1
4  1  1
5  2  1
6  0  2
7  1  2
8  2  2

ฉันต้องจัดเรียงตามลำดับxแรกและอันดับสองโดยy:

df2 = df.sort(["x", "y"])

   x  y
0  0  0
3  0  1
6  0  2
1  1  0
4  1  1
7  1  2
2  2  0
5  2  1
8  2  2

ฉันจะเปลี่ยนดัชนีให้มันขึ้นอีกครั้งได้อย่างไร ฉันจะได้รับสิ่งนี้อย่างไร:

   x  y
0  0  0
1  0  1
2  0  2
3  1  0
4  1  1
5  1  2
6  2  0
7  2  1
8  2  2

ฉันได้ลองสิ่งต่อไปนี้แล้ว น่าเสียดายที่มันไม่ได้เปลี่ยนดัชนีเลย:

df2.reindex(np.arange(len(df2.index)))

คุณสามารถรีเซ็ตดัชนีโดยใช้reset_indexเพื่อกลับดัชนีเริ่มต้นเป็น 0, 1, 2, ... , n-1 (และใช้drop=Trueเพื่อระบุว่าคุณต้องการทิ้งดัชนีที่มีอยู่แทนที่จะเพิ่มเป็นคอลัมน์เพิ่มเติมในดาต้าเฟรมของคุณ) :

In [19]: df2 = df2.reset_index(drop=True)

In [20]: df2
Out[20]:
   x  y
0  0  0
1  0  1
2  0  2
3  1  0
4  1  1
5  1  2
6  2  0
7  2  1
8  2  2

df.sort()เลิกใช้แล้วให้ใช้df.sort_values(...): https://pandas.pydata.org/pandas-docs/stable/generated/pandas.DataFrame.sort_values.html

จากนั้นทำตามคำตอบของ joris โดยทำ df.reset_index(drop=True)

เนื่องจาก pandas 1.0.0 df.sort_valuesมีพารามิเตอร์ใหม่ignore_indexที่ตอบสนองสิ่งที่คุณต้องการ:

In [1]: df2 = df.sort_values(by=['x','y'],ignore_index=True)

In [2]: df2
Out[2]:
   x  y
0  0  0
1  0  1
2  0  2
3  1  0
4  1  1
5  1  2
6  2  0
7  2  1
8  2  2

คุณสามารถตั้งค่าดัชนีใหม่ได้โดยใช้set_index:

df2.set_index(np.arange(len(df2.index)))

เอาท์พุต:

   x  y
0  0  0
1  0  1
2  0  2
3  1  0
4  1  1
5  1  2
6  2  0
7  2  1
8  2  2