在浮点数列表中仅将零更改为整数

Question

I have this output from my code

[0.0, 0.0, 0.25, 0.5, 1.0],

but I need to make it be like this -

[0, 0, 0.25, 0.5, 1.0]

Any tips on how to change the zeros to integers?

Answer 1

If you want to change all zeros to integers, a simple list comprehension will do.

>>> lst = [0.0, 0.0, 0.25, 0.5, 1.0]
>>> [x or 0 for x in lst]
[0, 0, 0.25, 0.5, 1.0]

If and only if x == 0 then the expression will be evaluated to 0 .

This is assuming no other falsy values like empty strings or lists are in lst .

Answer 2

This is a bit of a weird request, but you can just do it on a case by case basis, turning all zeros into int s:

>>> output = [0.0, 0.0, 0.25, 0.5, 1.0]
>>> [int(x) if not x else x for x in output]
[0, 0, 0.25, 0.5, 1.0]

That list comprehension is the same as

new_list=[]
for x in output:
    if x == 0:
        new_list.append(int(x))
    else:
        new_list.append(x)

The for -loop form is easier for making more complicated replacement rules if you need to tweak it.

Answer 3

You can simply do this by list comprehension :

Example :

If i want all the numbers that are less than 3 to be integers and rest float , then i would have done like this.

myList = [1.1 , 2.7 , 7.8 ,8 ,8.9 ,3.2]
myList = [ int(i) if i < 3 else i for i in myList ]
print(myList)

Output:

[1, 2, 7.8, 8, 8.9, 3.2]

Your required code is :

 myList = [0.0, 0.0, 0.25, 0.5, 1.0]
 myList = [int(i) if i == 0 else i for i in myList]
 print(myList)