将熊猫数据框转换为命名元组列表

Question

我正在寻找将pandas DataFrame转换为类型为NamedTuple的列表的最有效方法-以下是具有预期输出的简单示例。我想获得与数据帧中定义的类型对齐的正确类型转换。

from typing import NamedTuple
import pandas as pd

if __name__ == "__main__":
    data = [["tom", 10], ["nick", 15], ["juli", 14]]
    People = pd.DataFrame(data, columns=["Name", "Age"])
    Person = NamedTuple("Person", [("name", str), ("age", int)])
    # ...
    # ...
    # expected output
    # [Person(name='tom', age=10), Person(name='nick', age=15), Person(name='juli', age=14)]

Answer 1

用途：

a  = list(people.itertuples(index=False, name='Person'))
print(a)
[Person(Name='tom', Age=10), Person(Name='nick', Age=15), Person(Name='juli', Age=14)]

Answer 2

执行：

import pandas as pd

data = [["tom", 10], ["nick", 15], ["juli", 14]]
people = pd.DataFrame(data, columns=["Name", "Age"])

result  = list(people.itertuples(index=False))

print(result)