Question

我正在编写一个程序，它接受用户输入并将其保存在数组中。我目前正在尝试编写一个编辑数组中数据的函数。到目前为止，我有：

data = []
user_input = int(raw_input('How many Subject do you wish to add? : '))
for i in range (0, user_input):
    x = raw_input('Enter Subject: ')
    data.append(x)

print(data)

Answer 1

据我所知，我有这个代码给你。

data = []
user_input = int(raw_input('How many Subject do you wish to add? : '))
for i in range (0, user_input):
    x = raw_input('Enter Subject: ')
    data.append(x)

print(data)
yes = 1
no = 2
print "would you like to add/remove any items in list"
print "1: yes"
print "2: no"
edit_input = int(raw_input('choose?: '))
if edit_input == 1:
    edit_data = raw_input('Enter few more Subjects: ')
    data.append(edit_data)
else:
    edit_data = raw_input('Enter Subjects you want to remove: ')
    if edit_data in data:
        data.remove(edit_data)
    else:
        print "element not found"
print data

Answer 2

data = []
user_input = int(raw_input('How many Subject do you wish to add? : ')) 

for i in range (0, user_input): 
    x = raw_input('Enter Subject: ') 
    data.append(x) 

while True:
    print ("would you like to add/remove any items in list")
    print ("1: yes")
    print ("2: remove")
    print ("3: exit")
    edit_input = int(raw_input('choose?: '))
    if edit_input == 1:
        edit_data = raw_input('Enter a new Subject: ')
        data.append(edit_data)
    elif edit_input == 2:
        edit_data = raw_input('Enter a Subject you want to remove: ')
        if edit_data in data:
            data.remove(edit_data)
        else:
            print('Subject not found!')
    else:
        # breaking off the loop here
        break

print data

如何编写一个函数来编辑数组中的原始输入？

2 个答案: