read input from stdin in python code example

Example 1: input stdin python

import sys
data = sys.stdin.readline()
sys.stdout.write('Dive in')

Example 2: python read from stdin

import sys
data = sys.stdin.readlines()