Python 中实现类似C++的cin

突然发现牛客上有些题目的测试略坑Python。因为C++中默认用cin，所以‘ ’和‘\n’不同太区分。可是Python没这个功能啊。所以有时候发现程序通不过是跪在输入上。比如明明说好有3行的输入，结果实际测试中就只有2行，其中‘\n’偷懒用空格替代了。所以在想如何实现cin。求指点有没有个好想法解决这问题。我的解决时：用一个generator，然后调用next()读入。

import sys
def read():
	for line in sys.stdin:
		for s in line:
			yield s