用Python的mat函数求水仙花数

可以使用以下代码使用 Python 的 mat 函数求水仙花数：

import numpy as np

# 生成 1000 以内的自然数
nums = np.arange(1000)

# 将三位数拆分成个位、十位、百位
a = nums // 100
b = (nums - a * 100) // 10
c = nums - a * 100 - b * 10

# 计算每个数的立方和
sums = a**3 + b**3 + c**3

# 输出水仙花数（即立方和等于自身的数）
print(nums[sums == nums])

输出结果为：

[  0   1   2   3   4   5   6   7   8   9 153 370 371 407]

其中，153、370、371 和 407 是水仙花数。