data = ['cat', 'dog', 'None', 'Turtle', 'None']
new_data = []
for item in data:
if item == 'None':
new_data.append(data.index(item))
print new_data
>> [2,2]
How do I go about getting to this store new data as [2,4]? This is what I want. Thank you!
Use enumerate() while looping. This will track both, the current item and its index:
[index for index, x in enumerate(data) if x == "None"]
for idx, item in enumerate(data):
if item == 'None':
new_data.append(idx)
better yet, just use a list comprehension as in Sven's answer
Try:
In [1]: data = ['cat', 'dog', 'None', 'Turtle', 'None']
In [2]: [i for i,val in enumerate(data) if val == 'None']
Out[2]: [2, 4]
data.index(item) only returns the position of first occurance of the item in your list. You could simply do this:
for i in range(0,len(data)):
if data[i] == 'None':
new_data.append(i)
this should give you the required output
OR
check out Sven's answer
