Skip to content

Commit 05b8503

Browse files
code sample for pandas-dev#44900
1 parent 36611b8 commit 05b8503

File tree

1 file changed

+30
-0
lines changed

1 file changed

+30
-0
lines changed

bisect/44900.py

+30
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,30 @@
1+
# BUG: Using pd.concat(axis="columns") on differently sized MultiIndexed
2+
# DataFrames with a datetime index level containing exclusively NaT values
3+
# causes the level in the returned DataFrame to be a float instead of a datetime
4+
# #44900
5+
6+
import numpy as np
7+
8+
import pandas as pd
9+
10+
print(pd.__version__)
11+
df_a = pd.DataFrame({"a": range(5), "idx1": range(5), "idx2": [pd.NaT] * 5}).set_index(
12+
["idx1", "idx2"]
13+
)
14+
df_b = pd.DataFrame({"b": range(6), "idx1": range(6), "idx2": [pd.NaT] * 6}).set_index(
15+
["idx1", "idx2"]
16+
)
17+
result = pd.concat([df_a, df_b], axis="columns")
18+
print(result)
19+
20+
expected = pd.DataFrame(
21+
{
22+
"a": list(range(5)) + [np.nan],
23+
"b": range(6),
24+
"idx1": range(6),
25+
"idx2": [pd.NaT] * 6,
26+
}
27+
).set_index(["idx1", "idx2"])
28+
expected
29+
30+
pd.testing.assert_frame_equal(result, expected)

0 commit comments

Comments
 (0)