Search code examples
pythonraw-input

Using raw_input to define what variable list to search


How can i use the raw_input value stored in rowChoice be used in count? i.e. right now i define A.count. How would i wtite the correct equivanlent of rowChoice.count, where the user defines which of A,B,C,D to seach through.

A = ['available', 'unavailable','available','available','available']
B = ['unavailable', 'unavailable','available','available','available']
C = ['available', 'unavailable','unavailable','available','available']
D = ['available', 'unavailable','available','unavailable','unavailable']

rowChoice = raw_input('What row would you like to sit in? >> ')

count = A.count('available')
print'there are %r seats available in this row' %(count)

Solution

  • It would be best to use a dictionary for this:

    d = {'A': ['available', 'unavailable','available','available','available'],
         'B': ['unavailable', 'unavailable','available','available','available'],
         'C': ['available', 'unavailable','unavailable','available','available'],
         'D': ['available', 'unavailable','available','unavailable','unavailable']}
    
    letter = raw_input().upper()    
    d[letter].count('available')