Как упростить обработку списка в Python?

Question

Вот моя первая программа на Python, маленькая утилита, которая преобразует из восьмеричного кода Unix для прав доступа к файлам в символической форме:

s=raw_input("Octal? "); 
digits=[int(s[0]),int(s[1]),int(s[2])]; 
lookup=['','x','w','wx','r','rx','rw','rwx']; 
uout='u='+lookup[digits[0]]; 
gout='g='+lookup[digits[1]]; 
oout='o='+lookup[digits[2]]; 
print(uout+','+gout+','+oout); 

Есть ли способы сократить этот код, воспользоваться каким-то «обработкой списка «? Например, чтобы сразу применить функцию int ко всем трем символам s без необходимости явной индексации. И индексировать в lookup используя весь список digits сразу?

Answer 1

Вот немного оптимизированная версия кода:

s = raw_input("Octal? ") 
digits = map(int, s) 
lookup = ['','x','w','wx','r','rx','rw','rwx'] 
perms = [lookup[d] for d in digits] 
rights = ['{}={}'.format(*x) for x in zip('ugo', perms)] 
print ','.join(rights) 

Answer 2

digits=[int(s[0]),int(s[1]),int(s[2])]; 

можно записать в виде:

digits = map(int,s) 

или:

digits = [ int(x) for x in s ] #list comprehension 

Как это выглядит, как вы могли бы использовать python3.x (или планируете использовать его в будущем на основе ваше использование функции, например, использование печати), вы можете выбрать для понимания списка, если вы не захотите копать дальше, и используйте zip, как показано одним из более поздних ответов.

Answer 3

Вот моя трещина на него (в том числе и «-» для недостающих разрешений):

lookup = { 
    0b000 : '---', 
    0b001 : '--x', 
    0b010 : '-w-', 
    0b011 : '-wx', 
    0b100 : 'r--', 
    0b101 : 'r-x', 
    0b110 : 'rw-', 
    0b111 : 'rwx' 
} 

s = raw_input('octal?: ') 
print(','.join(# using ',' as the delimiter 
       r + '=' + lookup[int(n, 8)] # the letter followed by the permissions 
       for n, r in zip(tuple(s), 'ugo'))) # for each number/ letter pair 

Answer 4

Вы также можете сделать это с битмаски :

masks = { 
    0b100: 'r', # 4 
    0b010: 'x', # 2 
    0b001: 'w' # 1 
} 

octal = raw_input('Octal? ') 
result = '-' 

for digit in octal[1:]: 
    for mask, letter in sorted(masks.items(), reverse=True): 
     if int(digit, 8) & mask: 
      result += letter 
     else: 
      result += '-' 

print result 

Answer 5

Вот моя версия, вдохновленные решения в Blender:

bits = zip([4, 2, 1], "rwx") 
groups = "ugo" 

s = raw_input("Octal? "); 
digits = map(int, s) 

parts = [] 
for group, digit in zip(groups, digits): 
    letters = [letter for bit, letter in bits if digit & bit] 
    parts.append("{0}={1}".format(group, "".join(letters))) 

print ",".join(parts) 

Я думаю, что лучше не указывать в явном виде список lookup.