Skip to content

Commit 2c4db07

Browse files
add code sample for pandas-dev#36915
1 parent a41b502 commit 2c4db07

File tree

1 file changed

+20
-0
lines changed

1 file changed

+20
-0
lines changed

bisect/36915.py

+20
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,20 @@
1+
# BUG: Intersection of multiindex returns duplicates #36915
2+
3+
import pandas as pd
4+
5+
print(pd.__version__)
6+
7+
arraysA = [["val1", "val1", "val1", "val1"], ["val2", "val2", "val2", "val2"]]
8+
arraysB = [["val1"], ["val2"]]
9+
10+
indexA = pd.MultiIndex.from_arrays(arraysA, names=("idx1", "idx2"))
11+
indexB = pd.MultiIndex.from_arrays(arraysB, names=("idx1", "idx2"))
12+
13+
res = indexA.intersection(indexB)
14+
print(res)
15+
16+
expected = pd.MultiIndex.from_tuples([("val1", "val2")], names=["idx1", "idx2"])
17+
18+
from pandas import testing as tm
19+
20+
tm.assert_index_equal(res, expected)

0 commit comments

Comments
 (0)