how to use stdin instead of input python3 code example

Example: input stdin python

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