用于存储对象的 Python DataFrame 或列表

Question

我可以在 pandas/numpy Series-DataFrame/ndarray 中“存储”类的实例吗 就像我在列表中所做的那样？或者这些库支持内置类型（数字、字符串）。

例如，我有Point 和x,y 坐标，我想将Points 存储在Plane 中，这将返回给定坐标的Point。

#my class
class MyPoint:

    def __init__(self, x,y):
        self.x = x
        self.y = y

    @property
    def x(self):
        return self.x

    @property
    def y(self):
        return self.y

我在这里创建实例：

first_point = MyClass(1,1)
second_point = MyClass(2,2)

我可以将实例存储在某个列表中

my_list = []
my_list.append(first_point)
my_list.append(second_point)

列表中的问题是它的索引与 x,y 属性不对应。

字典/DataFrame 方法：

Plane = {"x" : [first_point.x, second_point.x], "y" : [first_point.y, second_point.y], "some_reference/id_to_point_instance" = ???}
Plane_pd = pd.DataFrame(Plane)

我读过帖子，在 DataFrame 中使用实例的“id”作为第三列值可能会导致垃圾收集器出现问题。

Answer 1

pandas.DataFrame 很乐意存储 python 对象。

一些测试代码来演示...

测试代码：

class MyPoint:
    def __init__(self, x, y):
        self._x = x
        self._y = y

    @property
    def x(self):
        return self._x

    @property
    def y(self):
        return self._y

my_list = [MyPoint(1, 1), MyPoint(2, 2)]
print(my_list)

plane_pd = pd.DataFrame([[p.x, p.y, p] for p in my_list],
                        columns=list('XYO'))
print(plane_pd.dtypes)
print(plane_pd)

结果：

[<__main__.MyPoint object at 0x033D2AF0>, <__main__.MyPoint object at 0x033D2B10>]

X     int64
Y     int64
O    object
dtype: object

   X  Y                                        O
0  1  1  <__main__.MyPoint object at 0x033D2AF0>
1  2  2  <__main__.MyPoint object at 0x033D2B10>

注意事项：

请注意，列表中的两个对象与数据框中的两个对象相同。还要注意O 列的dtype 是object。