如何編寫一個只接受內部帶有 int 或 float 類型的列表的 Python 函數？

Question

我需要編寫一個函數，該函數將采用 int 或 float 類型的列表。 如果有嵌套列表，函數必須返回“No”，如果只有 float 或 int 類型，函數必須返回“Yes”。 這是我的代碼

def take_list(lst):
    for i in lst:
        if isinstance(i, list) is True:
            return "No"
        elif isinstance(i, int) or isinstance(i, float):
            return "Yes"


print(take_list([1, 2, [3, 2]]))
print(take_list([1, 2, 3, 2]))

但這里的答案總是“是”。 我嘗試了“type”而不是“isinstance”，所以它不起作用

Answer 1

使用type怎么樣？

def take_list(lst):
    # if all(isinstance(item, (int, float)) for item in lst): # another way
    if all(type(item).__name__ in ['int', 'float'] for item in lst):
        return "Yes"
    else:
        return "No"


print(take_list([1, 2, [3, 2]]))     # No
print(take_list([1, 2, 3, 2, 1.2]))  # Yes

type是int和float類中的函數，使用___name___獲取類名稱。