Python中类似C的结构


444

有没有一种方法可以在Python中方便地定义类似C的结构?我讨厌写这样的东西:

class MyStruct():
    def __init__(self, field1, field2, field3):
        self.field1 = field1
        self.field2 = field2
        self.field3 = field3

5
半相关地,代数数据类型绝对是很棒的,但是要很好地使用它们,您通常需要模式匹配。
爱德华·Z·杨

50
除了编写繁琐之外,此方法还有什么问题吗?
levesque 2010年

2
您可能会发现dstruct有用: github.com/dorkitude/dstruct
Kyle Wild,

10
@levesque很难在没有错别字的情况下进行MyStruct = namedtuple("MyStruct", "field1 field2 field3")
重构

1
pandas.Series(a=42).a如果您的数据科学家应该这样做...
Mark Horvath

Answers:


341

使用命名的tuple,它已添加到Python 2.6的标准库的collections模块中。如果您需要支持Python 2.4,也可以使用Raymond Hettinger的命名元组配方。

这对于您的基本示例很好,但是也涵盖了以后可能会遇到的许多极端情况。您上面的片段将写为:

from collections import namedtuple
MyStruct = namedtuple("MyStruct", "field1 field2 field3")

新创建的类型可以这样使用:

m = MyStruct("foo", "bar", "baz")

您还可以使用命名参数:

m = MyStruct(field1="foo", field2="bar", field3="baz")

163
...但是namedtuple是不可变的。OP中的示例是可变的。
mhowison

27
@mhowison-就我而言,这只是一个优点。
ArtOfWarfare 2014年

3
不错的解决方案。您将如何遍历这些元组的数组?我假设字段1-3在元组对象之间必须具有相同的名称。
迈克尔·史密斯

2
namedtuple最多可以有四个参数,因此我们如何映射具有相应namedtuple的更多数据成员的结构
Kapil

3
@Kapil-namedtuple的第二个参数应该是成员名称的列表。该列表可以是任意长度。
ArtOfWarfare

222

更新:数据类

通过引入数据类的Python 3.7,我们非常接近。

下面的示例类似于下面的NamedTuple示例,但是结果对象是可变的,并且允许使用默认值。

from dataclasses import dataclass


@dataclass
class Point:
    x: float
    y: float
    z: float = 0.0


p = Point(1.5, 2.5)

print(p)  # Point(x=1.5, y=2.5, z=0.0)

如果您想使用更多特定的类型注释,那么这与新的键入模块配合使用非常好。

我一直在拼命等待!如果您问我,数据类和新的NamedTuple声明,再加上键入模块,真是天赐之物!

改进了NamedTuple声明

Python 3.6开始,只要您可以忍受不变性,它就会变得非常简单和美观(IMHO)。

一个声明NamedTuples的新方法被引入,它允许类型的注释,以及:

from typing import NamedTuple


class User(NamedTuple):
    name: str


class MyStruct(NamedTuple):
    foo: str
    bar: int
    baz: list
    qux: User


my_item = MyStruct('foo', 0, ['baz'], User('peter'))

print(my_item) # MyStruct(foo='foo', bar=0, baz=['baz'], qux=User(name='peter'))

6
伴侣,您刚刚度过了我的一天-一成不变的命令-谢谢:D
德米特里·阿基普琴科

10
dataclass模块是Python 3.7中的新增功能,但您可以pip install dataclasses。这是Python 3.6的反向端口。pypi.org/project/dataclasses/#description
Lavande

+1用于改进NamedTuple声明。如果您有几个变量,那么旧的方法实在令人不快阅读...
gebbissimo

@Lavande我可以知道在3.6和3.7之间发生了什么重大变化时,您必须向后移植一个次要版本...吗?
紫色冰

1
@PurpleIce这是PEP 557的实现,数据类@dataclass的细节在这里:pypi.org/project/dataclasses/#description
Lavande

96

您可以在很多情况下使用元组,而在C中使用结构(例如x,y坐标或RGB颜色)。

对于其他所有内容,您都可以使用字典或类似这样的实用程序类:

>>> class Bunch:
...     def __init__(self, **kwds):
...         self.__dict__.update(kwds)
...
>>> mystruct = Bunch(field1=value1, field2=value2)

我认为“确定性”讨论在此处,在Python Cookbook的发行版本中。


5
空的教室会做同样的事情吗?
Kurt Liu

44
请注意,如果您是python的新手,则元组一旦创建便是只读的,这与C结构不同
LeBleu 2011年

2
@KurtLiu不,它可能会说TypeError: this constructor takes no arguments
Evgeni Sergeev

84

也许您正在寻找没有构造函数的Structs:

class Sample:
  name = ''
  average = 0.0
  values = None # list cannot be initialized here!


s1 = Sample()
s1.name = "sample 1"
s1.values = []
s1.values.append(1)
s1.values.append(2)
s1.values.append(3)

s2 = Sample()
s2.name = "sample 2"
s2.values = []
s2.values.append(4)

for v in s1.values:   # prints 1,2,3 --> OK.
  print v
print "***"
for v in s2.values:   # prints 4 --> OK.
  print v

5
从技术上讲,您在这里所做的工作是可行的,但对于许多用户而言,为什么行之有效可能并不立刻显而易见。您的声明class Sample:不会立即执行任何操作;他们设置类的属性。那些总是可以例如访问Sample.name
Channing Moore

22
什么你实际上做的是将实例属性的对象s1,并s2在运行时。除非另有禁止,否则您可以name随时在任何类的任何实例上添加或修改属性,无论该类是否具有name属性。这样做的最大功能问题可能是同一类的不同实例的行为将取决于您是否设置了name。如果进行更新Sample.name,则任何没有显式设置name属性的对象都将返回新对象name
Channing Moore

2
这与结构差不多—简短的“类”,不带方法,“字段”(类属性,我知道),带有默认值。只要它不是可变类型(dict,列表),就可以了。当然,您可以使用PyCharm的“类没有初始化方法” 来对抗PEP-8或“友好” IDE检查。
Tomasz Gandor

4
我尝试了Channing Moore描述的副作用。self如果您问我,不值得花几个关键词和一个构造函数来节省时间。如果Jose可以编辑其答案以添加警告消息,说明在实例之间意外共享值的风险,我将不胜感激。
斯特凡C.

@ChanningMoore:我试图重新创建您正在描述的问题,但是失败了。您能否提出一个最小的工作示例,以解决此问题?
gebbissimo

67

字典怎么样?

像这样:

myStruct = {'field1': 'some val', 'field2': 'some val'}

然后,您可以使用它来操纵值:

print myStruct['field1']
myStruct['field2'] = 'some other values'

并且值不必是字符串。它们几乎可以是任何其他对象。


34
这也是我的方法,但是我觉得这很危险,因为字典可以接受任何东西作为键。如果我打算设置myStruct [“ field”],则设置myStruct [“ ffield”]不会出错。当我稍后使用或重新使用myStruct [“ field”]时,该问题可能(也可能不会)变得明显。我喜欢PabloG的方法。
mobabo 2014年

PabloG的存在相同的问题。尝试在他的代码中添加以下代码:pt3.w = 1 print pt3.w 在具有字典的语言中,最好使用它们,尤其是对于要序列化的对象,因为只要您不奇怪,就可以自动使用import json保存它们和其他序列化库。字典中的内容。Dicts是保持数据和逻辑分离的解决方案,并且比那些不想编写自定义序列化和反序列化函数并且不想使用pickle这样的非便携式序列化程序的人的结构更好。
Poikilos

27

dF:太酷了……我不知道我可以使用dict访问类中的字段。

马克:我希望我遇到的情况恰好是当我想要一个元组,却又没有字典那么重的时候。

您可以使用字典访问类的字段,因为类的字段,其方法及其所有属性都使用dict在内部存储(至少在CPython中)。

...这将引导我们提出您的第二条评论。相信Python字典是“繁重的”是一个极端的非Python的概念。阅读此类评论会杀死我的Python Zen。这不好。

您会看到,在声明一个类时,实际上是在围绕字典创建一个非常复杂的包装器-因此,如果有的话,与使用简单的字典相比,您将增加更多的开销。顺便说一句,开销在任何情况下都是没有意义的。如果您正在处理对性能至关重要的应用程序,请使用C或类似的东西。


5
#1,Cython!= CPython。我认为您是在谈论CPython,这是用C编写的Python的实现,而不是Cython,Cython是一个将Python代码交叉编译为C代码的项目。我修改了您的答案以解决此问题。#2,我认为当他说字典很重时,他指的是语法。self['member']比的长度长3个字符self.member,而且这些字符都相对不适合手腕使用。
ArtOfWarfare 2015年

19

您可以将标准库中可用的C结构子类化。该ctypes的模块提供了一个结构类。来自文档的示例:

>>> from ctypes import *
>>> class POINT(Structure):
...     _fields_ = [("x", c_int),
...                 ("y", c_int)]
...
>>> point = POINT(10, 20)
>>> print point.x, point.y
10 20
>>> point = POINT(y=5)
>>> print point.x, point.y
0 5
>>> POINT(1, 2, 3)
Traceback (most recent call last):
  File "<stdin>", line 1, in ?
ValueError: too many initializers
>>>
>>> class RECT(Structure):
...     _fields_ = [("upperleft", POINT),
...                 ("lowerright", POINT)]
...
>>> rc = RECT(point)
>>> print rc.upperleft.x, rc.upperleft.y
0 5
>>> print rc.lowerright.x, rc.lowerright.y
0 0
>>>

18

我还想添加一个使用slot的解决方案:

class Point:
    __slots__ = ["x", "y"]
    def __init__(self, x, y):
        self.x = x
        self.y = y

一定要查看文档中的插槽,但是插槽的快速说明是python的一种说法:“如果您可以将这些属性以及仅这些属性锁定到类中,以致您承诺一旦该类就不会添加任何新属性实例化(是的,您可以向类实例添加新属性,请参见下面的示例),然后我将取消大的内存分配,该内存分配允许向类实例添加新属性,并仅使用我需要的这些插槽化属性。”

向类实例添加属性的示例(因此不使用插槽):

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

p1 = Point(3,5)
p1.z = 8
print(p1.z)

输出8

尝试向使用插槽的类实例添加属性的示例:

class Point:
    __slots__ = ["x", "y"]
    def __init__(self, x, y):
        self.x = x
        self.y = y

p1 = Point(3,5)
p1.z = 8

输出:AttributeError:'Point'对象没有属性'z'

这可以有效地用作结构,并且比类使用更少的内存(就像结构一样,尽管我没有确切研究多少)。如果您将创建大量对象实例并且不需要添加属性,则建议使用插槽。一个点对象就是一个很好的例子,因为可能实例化许多点来描述一个数据集。


17

您还可以按位置将init参数传递给实例变量

# Abstract struct class       
class Struct:
    def __init__ (self, *argv, **argd):
        if len(argd):
            # Update by dictionary
            self.__dict__.update (argd)
        else:
            # Update by position
            attrs = filter (lambda x: x[0:2] != "__", dir(self))
            for n in range(len(argv)):
                setattr(self, attrs[n], argv[n])

# Specific class
class Point3dStruct (Struct):
    x = 0
    y = 0
    z = 0

pt1 = Point3dStruct()
pt1.x = 10

print pt1.x
print "-"*10

pt2 = Point3dStruct(5, 6)

print pt2.x, pt2.y
print "-"*10

pt3 = Point3dStruct (x=1, y=2, z=3)
print pt3.x, pt3.y, pt3.z
print "-"*10

7
按位置更新将忽略属性的声明顺序,而是使用其字母排序。因此,如果您更改Point3dStruct声明中的行顺序,Point3dStruct(5, 6)将无法按预期工作。奇怪的是,这六年来没有人写过这篇文章。
青金石

可以在您的出色代码中添加Python 3版本吗?做得好!我喜欢您将一些抽象的东西,并使其与第二个特定的类明确。这对于错误处理/捕获应该是好的。对于Python 3,只需更改print> print(),然后attrs[n]> next(attrs)(过滤器现在是它自己的可迭代对象,并且需要next)。
乔纳森·科玛

10

每当我需要一个“行为也像字典的即时数据对象”(我认为C结构!)时,我就会想到这个可爱的技巧:

class Map(dict):
    def __init__(self, **kwargs):
        super(Map, self).__init__(**kwargs)
        self.__dict__ = self

现在您可以说:

struct = Map(field1='foo', field2='bar', field3=42)

self.assertEquals('bar', struct.field2)
self.assertEquals(42, struct['field3'])

当您需要一个“不是类的数据包”以及namedtuple难以理解时,非常方便。


我使用pandas.Series(a = 42);-)
Mark Horvath

8

您可以通过以下方式在python中访问C-Style结构。

class cstruct:
    var_i = 0
    var_f = 0.0
    var_str = ""

如果您只想使用cstruct的对象

obj = cstruct()
obj.var_i = 50
obj.var_f = 50.00
obj.var_str = "fifty"
print "cstruct: obj i=%d f=%f s=%s" %(obj.var_i, obj.var_f, obj.var_str)

如果要创建cstruct的对象数组

obj_array = [cstruct() for i in range(10)]
obj_array[0].var_i = 10
obj_array[0].var_f = 10.00
obj_array[0].var_str = "ten"

#go ahead and fill rest of array instaces of struct

#print all the value
for i in range(10):
    print "cstruct: obj_array i=%d f=%f s=%s" %(obj_array[i].var_i, obj_array[i].var_f, obj_array[i].var_str)

注意:请使用您的结构名称而不是“ cstruct”名称,而不要使用var_i,var_f,var_str,请定义结构的成员变量。



8

这里的一些答案非常详尽。我找到的最简单的选项是(来自:http : //norvig.com/python-iaq.html):

class Struct:
    "A structure that can have any fields defined."
    def __init__(self, **entries): self.__dict__.update(entries)

初始化:

>>> options = Struct(answer=42, linelen=80, font='courier')
>>> options.answer
42

增加更多:

>>> options.cat = "dog"
>>> options.cat
dog

编辑:对不起,没有看到这个例子。


5

这可能有点晚了,但是我使用Python Meta-Classes(也是下面的装饰器版本)提出了一个解决方案。

什么时候 __init__在运行时被调用时,它抓住每个参数和它们的值,并将它们分配为实例变量上您的课。这样,您可以制作类似结构的类,而不必手动分配每个值。

我的示例没有错误检查,因此更容易理解。

class MyStruct(type):
    def __call__(cls, *args, **kwargs):
        names = cls.__init__.func_code.co_varnames[1:]

        self = type.__call__(cls, *args, **kwargs)

        for name, value in zip(names, args):
            setattr(self , name, value)

        for name, value in kwargs.iteritems():
            setattr(self , name, value)
        return self 

它在起作用。

>>> class MyClass(object):
    __metaclass__ = MyStruct
    def __init__(self, a, b, c):
        pass


>>> my_instance = MyClass(1, 2, 3)
>>> my_instance.a
1
>>> 

将其发布在reddit上/ u / matchu发布了一个更干净的装饰器版本。除非您要扩展元类版本,否则我建议您使用它。

>>> def init_all_args(fn):
    @wraps(fn)
    def wrapped_init(self, *args, **kwargs):
        names = fn.func_code.co_varnames[1:]

        for name, value in zip(names, args):
            setattr(self, name, value)

        for name, value in kwargs.iteritems():
            setattr(self, name, value)

    return wrapped_init

>>> class Test(object):
    @init_all_args
    def __init__(self, a, b):
        pass


>>> a = Test(1, 2)
>>> a.a
1
>>> 

该死的-今天我花了两个小时编写自己的装饰器,然后才发现。无论如何,发布我的,因为它处理默认值而您的不处理。stackoverflow.com/a/32448434/901641
ArtOfWarfare 2015年

+1代表提及func_code。开始朝那个方向挖掘,在那里发现了很多有趣的东西。
wombatonfire '16

5

我编写了一个装饰器,可以将其用于任何方法,以便将传入的所有参数或任何默认值分配给该实例。

def argumentsToAttributes(method):
    argumentNames = method.func_code.co_varnames[1:]

    # Generate a dictionary of default values:
    defaultsDict = {}
    defaults = method.func_defaults if method.func_defaults else ()
    for i, default in enumerate(defaults, start = len(argumentNames) - len(defaults)):
        defaultsDict[argumentNames[i]] = default

    def newMethod(self, *args, **kwargs):
        # Use the positional arguments.
        for name, value in zip(argumentNames, args):
            setattr(self, name, value)

        # Add the key word arguments. If anything is missing, use the default.
        for name in argumentNames[len(args):]:
            setattr(self, name, kwargs.get(name, defaultsDict[name]))

        # Run whatever else the method needs to do.
        method(self, *args, **kwargs)

    return newMethod

快速演示。请注意,我使用位置参数a,使用默认值b和命名参数c。然后self,我打印所有3个引用,以显示在输入方法之前已正确分配了它们。

class A(object):
    @argumentsToAttributes
    def __init__(self, a, b = 'Invisible', c = 'Hello'):
        print(self.a)
        print(self.b)
        print(self.c)

A('Why', c = 'Nothing')

请注意,我的装饰器应使用任何方法,而不仅仅是__init__


5

我在这里看不到这个答案,因此我想添加一下,因为我现在倾向于使用Python并发现了它。所述的Python教程(Python 2中在这种情况下)给出以下简单而有效的实施例:

class Employee:
    pass

john = Employee()  # Create an empty employee record

# Fill the fields of the record
john.name = 'John Doe'
john.dept = 'computer lab'
john.salary = 1000

即,创建一个空的类对象,然后实例化该字段,并动态添加字段。

这样做的好处是非常简单。缺点是它不是特别自我记录(预期的成员未在“定义”类中的任何位置列出),并且未设置的字段在访问时会引起问题。这两个问题可以通过以下方法解决:

class Employee:
    def __init__ (self):
        self.name = None # or whatever
        self.dept = None
        self.salary = None

现在,您至少可以一眼看出程序将期望哪些字段。

两者都容易出现错别字,john.slarly = 1000会成功。仍然可以。


4

这是一个使用类(从未实例化)保存数据的解决方案。我喜欢这种方式,几乎不需要打字,也不需要任何其他软件包

class myStruct:
    field1 = "one"
    field2 = "2"

您以后可以根据需要添加更多字段:

myStruct.field3 = 3

要获取值,请照常访问这些字段:

>>> myStruct.field1
'one'

2

我个人也喜欢这个变体。它扩展了@dF的答案

class struct:
    def __init__(self, *sequential, **named):
        fields = dict(zip(sequential, [None]*len(sequential)), **named)
        self.__dict__.update(fields)
    def __repr__(self):
        return str(self.__dict__)

它支持两种初始化模式(可以混合使用):

# Struct with field1, field2, field3 that are initialized to None.
mystruct1 = struct("field1", "field2", "field3") 
# Struct with field1, field2, field3 that are initialized according to arguments.
mystruct2 = struct(field1=1, field2=2, field3=3)

此外,它的打印效果更好:

print(mystruct2)
# Prints: {'field3': 3, 'field1': 1, 'field2': 2}

2

以下对结构的解决方案的灵感来自namedtuple实现和一些先前的答案。但是,与namedtuple不同,它的值是可变的,但是就像名称/属性中不可变的c样式结构一样,而普通的类或dict则不是。

_class_template = """\
class {typename}:
def __init__(self, *args, **kwargs):
    fields = {field_names!r}

    for x in fields:
        setattr(self, x, None)            

    for name, value in zip(fields, args):
        setattr(self, name, value)

    for name, value in kwargs.items():
        setattr(self, name, value)            

def __repr__(self):
    return str(vars(self))

def __setattr__(self, name, value):
    if name not in {field_names!r}:
        raise KeyError("invalid name: %s" % name)
    object.__setattr__(self, name, value)            
"""

def struct(typename, field_names):

    class_definition = _class_template.format(
        typename = typename,
        field_names = field_names)

    namespace = dict(__name__='struct_%s' % typename)
    exec(class_definition, namespace)
    result = namespace[typename]
    result._source = class_definition

    return result

用法:

Person = struct('Person', ['firstname','lastname'])
generic = Person()
michael = Person('Michael')
jones = Person(lastname = 'Jones')


In [168]: michael.middlename = 'ben'
Traceback (most recent call last):

  File "<ipython-input-168-b31c393c0d67>", line 1, in <module>
michael.middlename = 'ben'

  File "<string>", line 19, in __setattr__

KeyError: 'invalid name: middlename'

2

有一个专门用于此目的的python包。见cstruct2py

cstruct2py是一个纯Python库,用于从C代码生成python类,并使用它们来打包和解压缩数据。该库可以解析C头(结构,联合,枚举和数组声明),并在python中进行仿真。生成的pythonic类可以解析和打包数据。

例如:

typedef struct {
  int x;
  int y;
} Point;

after generating pythonic class...
p = Point(x=0x1234, y=0x5678)
p.packed == "\x34\x12\x00\x00\x78\x56\x00\x00"

如何使用

首先,我们需要生成pythonic结构:

import cstruct2py
parser = cstruct2py.c2py.Parser()
parser.parse_file('examples/example.h')

现在我们可以从C代码导入所有名称:

parser.update_globals(globals())

我们也可以直接这样做:

A = parser.parse_string('struct A { int x; int y;};')

从C代码使用类型和定义

a = A()
a.x = 45
print a
buf = a.packed
b = A(buf)
print b
c = A('aaaa11112222', 2)
print c
print repr(c)

输出将是:

{'x':0x2d, 'y':0x0}
{'x':0x2d, 'y':0x0}
{'x':0x31316161, 'y':0x32323131}
A('aa111122', x=0x31316161, y=0x32323131)

克隆

对于克隆cstruct2py运行:

git clone https://github.com/st0ky/cstruct2py.git --recursive

0

我认为Python结构字典适合此要求。

d = dict{}
d[field1] = field1
d[field2] = field2
d[field2] = field3

0

https://stackoverflow.com/a/32448434/159695在Python3中不起作用。

https://stackoverflow.com/a/35993/159695可在Python3中使用。

我将其扩展为添加默认值。

class myStruct:
    def __init__(self, **kwds):
        self.x=0
        self.__dict__.update(kwds) # Must be last to accept assigned member variable.
    def __repr__(self):
        args = ['%s=%s' % (k, repr(v)) for (k,v) in vars(self).items()]
        return '%s(%s)' % ( self.__class__.__qualname__, ', '.join(args) )

a=myStruct()
b=myStruct(x=3,y='test')
c=myStruct(x='str')

>>> a
myStruct(x=0)
>>> b
myStruct(x=3, y='test')
>>> c
myStruct(x='str')

0

如果@dataclass没有3.7,并且需要可变性,则以下代码可能对您有用。它具有很好的自我说明性和IDE友好性(自动完成),可防止重复编写两次,易于扩展,并且测试所有实例变量是否已完全初始化非常简单:

class Params():
    def __init__(self):
        self.var1 : int = None
        self.var2 : str = None

    def are_all_defined(self):
        for key, value in self.__dict__.items():
            assert (value is not None), "instance variable {} is still None".format(key)
        return True


params = Params()
params.var1 = 2
params.var2 = 'hello'
assert(params.are_all_defined)

0

这是一个快速而肮脏的把戏:

>>> ms = Warning()
>>> ms.foo = 123
>>> ms.bar = 'akafrit'

如何运作?它只是重复使用内置类Warning(从派生Exception),并使用它,因为它是您自己定义的类。

优点是您不需要首先导入或定义任何内容,“警告”是一个简短的名称,并且还可以清楚地表明您正在做一些肮脏的事情,除了您的小型脚本之外,其他任何地方都不应使用。

顺便说一句,我试图找到一些更简单的东西,ms = object()但是没有(最后一个例子不起作用)。如果您有一个,我很感兴趣。


0

我发现做到这一点的最佳方法是使用自定义词典类,如本文中所述:https : //stackoverflow.com/a/14620633/8484485

如果需要iPython自动补全支持,只需定义dir()函数,如下所示:

class AttrDict(dict):
    def __init__(self, *args, **kwargs):
        super(AttrDict, self).__init__(*args, **kwargs)
        self.__dict__ = self
    def __dir__(self):
        return self.keys()

然后,您可以像这样定义伪结构:(此嵌套)

my_struct=AttrDict ({
    'com1':AttrDict ({
        'inst':[0x05],
        'numbytes':2,
        'canpayload':False,
        'payload':None
    })
})

然后,您可以像这样访问my_struct内的值:

print(my_struct.com1.inst)

=>[5]


0

NamedTuple很舒服。但没有人共享性能和存储空间。

from typing import NamedTuple
import guppy  # pip install guppy
import timeit


class User:
    def __init__(self, name: str, uid: int):
        self.name = name
        self.uid = uid


class UserSlot:
    __slots__ = ('name', 'uid')

    def __init__(self, name: str, uid: int):
        self.name = name
        self.uid = uid


class UserTuple(NamedTuple):
    # __slots__ = ()  # AttributeError: Cannot overwrite NamedTuple attribute __slots__
    name: str
    uid: int


def get_fn(obj, attr_name: str):
    def get():
        getattr(obj, attr_name)
    return get
if 'memory test':
    obj = [User('Carson', 1) for _ in range(1000000)]      # Cumulative: 189138883
    obj_slot = [UserSlot('Carson', 1) for _ in range(1000000)]          # 77718299  <-- winner
    obj_namedtuple = [UserTuple('Carson', 1) for _ in range(1000000)]   # 85718297
    print(guppy.hpy().heap())  # Run this function individually. 
    """
    Index  Count   %     Size   % Cumulative  % Kind (class / dict of class)
     0 1000000    24 112000000 34 112000000  34 dict of __main__.User
     1 1000000    24 64000000  19 176000000  53 __main__.UserTuple
     2 1000000    24 56000000  17 232000000  70 __main__.User
     3 1000000    24 56000000  17 288000000  87 __main__.UserSlot
     ...
    """

if 'performance test':
    obj = User('Carson', 1)
    obj_slot = UserSlot('Carson', 1)
    obj_tuple = UserTuple('Carson', 1)

    time_normal = min(timeit.repeat(get_fn(obj, 'name'), repeat=20))
    print(time_normal)  # 0.12550550000000005

    time_slot = min(timeit.repeat(get_fn(obj_slot, 'name'), repeat=20))
    print(time_slot)  # 0.1368690000000008

    time_tuple = min(timeit.repeat(get_fn(obj_tuple, 'name'), repeat=20))
    print(time_tuple)  # 0.16006120000000124

    print(time_tuple/time_slot)  # 1.1694481584580898  # The slot is almost 17% faster than NamedTuple on Windows. (Python 3.7.7)

如果您__dict__不使用,请在__slots__(更高的性能和存储空间)和NamedTuple(便于阅读和使用)之间进行选择

您可以查看这个链接(的用法 ),以获得更多的__slots__信息。

By using our site, you acknowledge that you have read and understand our Cookie Policy and Privacy Policy.
Licensed under cc by-sa 3.0 with attribution required.