将浮点数转换为数组中的int

Question

我有一个形状为（0,5）的数组。 内容为[[4.0, 3.0, 0.0, 1.0, 5.0]]

如何将其转换为[[4 3 0 1 5]]？

Answer 1

一个简单的列表理解就可以做到这一点。

a = [[4.0, 3.0, 0.0, 1.0, 5.0]]
b = [[int(x) for x in a[0]]]

我们要求Python将列表a[0]的每个元素转换为int。查找list comprehensions，以获取有关其工作方式以及以后如何自己做的更多信息。