如果Python函数根据参数返回不同的类型,这是一种好的代码样式吗? - python

如果Python函数根据所提供的参数返回不同的类型,这是好的代码样式吗?

def foo(bar):
  if bar is None:
    return None
  elif bar == 1:
    return 1*1
  else:
    return [b*b for b in bar]

如果bar为None,则foo返回None

foo如果1返回bar == 1

如果bar是一个元组/整数列表,则foo返回一个int列表

例子:

>> foo(None)
None
>> foo(1)
1
>> foo(1, 2, 3, 4)
[1, 4, 9, 16]

返回Noneint应该可以,但是根据函数参数返回intint列表是否可以?您可能会说这是可以的,因为用户知道期望使用哪种返回类型并且不需要类型检查(在这种情况下,我会说这不行),但是有人可能会说将函数拆分为更好两个函数,一个期望一个int并返回一个int,另一个期望一个int列表并返回一个int列表。

参考方案

这完全取决于您的用例。这是标准库中的示例,其中结果的类型取决于输入的类型:

>>> import operator
>>> operator.add(1, 2)
3
>>> operator.add(1.0, 2.0)
3.0

这种行为通常可以,并且可以用@typing.overload记录。

这是一个示例,其中结果的类型取决于输入的值:

>>> import json
>>> json.loads('1')
1
>>> json.loads('[1]')
[1]

这种类型的行为通常保留给序列化和反序列化,或者模糊类型/值边界的API,例如astype中的np.int_(3).astype(bool)

另一方面,这是一个明显设计不良的函数示例:

from typing import Union  # make sure to document the mixed return type

def is_even(x: int) -> Union[bool, str]:
    if x % 2 == 0:
        return True
    else:
        return "no"

在不了解您的特定用例的情况下,很难在此处提供建议。

Python pytz时区函数返回的时区为9分钟 - python

由于某些原因,我无法从以下代码中找出原因:>>> from pytz import timezone >>> timezone('America/Chicago') 我得到:<DstTzInfo 'America/Chicago' LMT-1 day, 18:09:00 STD…

Python:如何从字节中提取特定位? - python

我有一条消息,显示为14 09 00 79 3d 00 23 27。我可以通过调用message[4]从此消息中提取每个字节,例如,这将给我3d。如何从该字节中提取单个8位?例如,如何将24-27位作为单个消息?只需28位? 参考方案 要回答问题的第二部分,您可以使用按位运算来获取特定的位值# getting your message as int i = …

Python:检查是否存在维基百科文章 - python

我试图弄清楚如何检查Wikipedia文章是否存在。例如,https://en.wikipedia.org/wiki/Food 存在,但是https://en.wikipedia.org/wiki/Fod 不会,页面只是说:“维基百科没有此名称的文章。”谢谢! 参考方案 >>> import urllib >>> prin…

Python GPU资源利用 - python

我有一个Python脚本在某些深度学习模型上运行推理。有什么办法可以找出GPU资源的利用率水平?例如,使用着色器,float16乘法器等。我似乎在网上找不到太多有关这些GPU资源的文档。谢谢! 参考方案 您可以尝试在像Renderdoc这样的GPU分析器中运行pyxthon应用程序。它将分析您的跑步情况。您将能够获得有关已使用资源,已用缓冲区,不同渲染状态上…

在返回'Response'(Python)中传递多个参数 - python

我在Angular工作,正在使用Http请求和响应。是否可以在“响应”中发送多个参数。角度文件:this.http.get("api/agent/applicationaware").subscribe((data:any)... python文件:def get(request): ... return Response(seriali…