1. 程式人生 > >Python: 在不確定需要輸入行數的情況下,逐行讀取輸入資料

Python: 在不確定需要輸入行數的情況下,逐行讀取輸入資料

HACKER RANK的一個程式設計題
import sys 

# Read input and assemble Phone Book
n = int(input())
phoneBook = {}
for i in range(n):
    contact = input().split(' ')
    phoneBook[contact[0]] = contact[1]

# Process Queries
lines = sys.stdin.readlines()
for i in lines:
    name = i.strip()
    if name in phoneBook:
        print(name + '=' + str( phoneBook[name] ))
    else:
        print('Not found')

使用sys模組的stdin.readlines可以讀取螢幕上輸入的所有行,在使用ctrl+d的時候結束。