update code sample for pandas-dev#43206

simonjayhawkins · simonjayhawkins · commit 7a404a7a4adf · 2021-08-25T14:13:25.000+01:00
diff --git a/bisect/43206.py b/bisect/43206.py
@@ -22,12 +22,17 @@
     }
 )
 
-df1 = df.groupby("uid", as_index=False)[["uid", "str_val", "date_val"]].apply(
+# df1 = df.groupby("uid", as_index=False)[["uid", "str_val", "date_val"]].apply(
+#     lambda x: x.sort_values(by="str_val", ascending=True)
+# )
+# print(df1)
+
+df2 = df.groupby("uid", as_index=False)[["uid", "str_val"]].apply(
     lambda x: x.sort_values(by="str_val", ascending=True)
 )
-print(df1)
+print(df2)
 
 expected = pd.MultiIndex.from_tuples(
     [(0, 0), (0, 2), (1, 1), (1, 3)],
 )
-pd.testing.assert_index_equal(df1.index, expected)
+pd.testing.assert_index_equal(df2.index, expected)

Original file line number	Diff line number	Diff line change
`@@ -22,12 +22,17 @@`
`22`	`22`	`}`
`23`	`23`	`)`
`24`	`24`
`25`		`-df1 = df.groupby("uid", as_index=False)[["uid", "str_val", "date_val"]].apply(`
	`25`	`+# df1 = df.groupby("uid", as_index=False)[["uid", "str_val", "date_val"]].apply(`
	`26`	`+# lambda x: x.sort_values(by="str_val", ascending=True)`
	`27`	`+# )`
	`28`	`+# print(df1)`
	`29`	`+`
	`30`	`+df2 = df.groupby("uid", as_index=False)[["uid", "str_val"]].apply(`
`26`	`31`	`lambda x: x.sort_values(by="str_val", ascending=True)`
`27`	`32`	`)`
`28`		`-print(df1)`
	`33`	`+print(df2)`
`29`	`34`
`30`	`35`	`expected = pd.MultiIndex.from_tuples(`
`31`	`36`	`[(0, 0), (0, 2), (1, 1), (1, 3)],`
`32`	`37`	`)`
`33`		`-pd.testing.assert_index_equal(df1.index, expected)`
	`38`	`+pd.testing.assert_index_equal(df2.index, expected)`