Yes, an NRI can purchase shares of an Indian company on a stock exchange in India, under the portfolio investment scheme on repatriation and/or non-repatriation basis. A PIO is considered as an NRI and hence all the benefits / conditions apply accordingly.