Python:有条件地修改用户输入

时间:2016-02-20 12:00:52

标签: python-3.4

我有一个.txt,我搜索的数据是这样形成的:' nnnnn'其中n =整数。

为此,我搜索:

with open("path of file to search") as search:
    num = input ("data : ")

如果用户输入具有相同的格式' nnnnn',则此方法可以正常工作。我的问题是输入可能会有所不同:' nnnnn'到了' n'和' MM0nnnnn'(文件中的数据总是长度为5,从'0000n'到' nnnnn')。

因此,我想在输入中添加两个条件:

  1. 如果输入中的MM0不考虑
  2. ,而输入长度为< 5向左添加零
  3. 我试过但没有成功:

    表示1:

    if 'MM0' in num :
            num.replace('MM0','')
    

    for 2:

    while len(num) < 5:
        num.append('0','0')
    

    我在网上看过这个,但我当然错了。

1 个答案:

答案 0 :(得分:0)

我能够解决这个问题:

for line in search:

        if 'MM0' in str(num) and str(num).replace('MM0','') in line: #Remove MM0 if present in input
            print (line)


        if str(num).zfill(5) in line:  #Fill with zeros to the left if input <5
            print(line)

它可以帮助那里的人。