def test_from_dicts_struct() -> None:
assert pl.from_dicts([{
"a": 1,
"b": {
"a": 1,
"b": 2
}
}]).to_series(1).to_list() == [{
"a": 1,
"b": 2
}]
assert pl.from_dicts([{
"a": 1,
"b": {
"a_deep": 1,
"b_deep": {
"a_deeper": [1, 2, 4]
}
}
}]).to_series(1).to_list() == [{
"a_deep": 1,
"b_deep": {
"a_deeper": [1, 2, 4]
}
}]
data = [
{
"a": [{
"b": 0,
"c": 1
}]
},
{
"a": [{
"b": 1,
"c": 2
}]
},
]
assert pl.from_dicts(data).to_series().to_list() == [
[{
"b": 0,
"c": 1
}],
[{
"b": 1,
"c": 2
}],
]
def test_from_dicts_list_without_dtype() -> None:
assert pl.from_dicts([{
"id": 1,
"hint": ["some_text_here"]
}, {
"id": 2,
"hint": [None]
}]).to_dict(False) == {
"id": [1, 2],
"hint": [["some_text_here"], [None]]
}
def test_issue_2529() -> None:
def stdize_out(value: str, control_for: str) -> pl.Expr:
return (pl.col(value) - pl.mean(value).over(control_for)
) / pl.std(value).over(control_for)
df = pl.from_dicts([{
"cat": cat,
"val1": cat + _,
"val2": cat + _
} for cat in range(2) for _ in range(2)])
out = df.select([
"*",
stdize_out("val1", "cat").alias("out1"),
stdize_out("val2", "cat").alias("out2"),
])
assert out["out1"].to_list() == out["out2"].to_list()
def test_from_dicts_struct() -> None:
data = [{"a": {"b": 1, "c": 2}, "d": 5}, {"a": {"b": 3, "c": 4}, "d": 6}]
df = pl.from_dicts(data)
assert df.shape == (2, 2)
assert df["a"][0] == (1, 2)
assert df["a"][1] == (3, 4)
def test_from_dicts() -> None:
data = [{"a": 1, "b": 4}, {"a": 2, "b": 5}, {"a": 3, "b": 6}]
df = pl.from_dicts(data)
assert df.shape == (3, 2)