{"cells":[{"cell_type":"markdown","metadata":{"_cell_guid":"b878500d-ac91-574f-fb85-af99a66ff166"},"source":"# Reading WKT and plotting with Pyplot"},{"cell_type":"code","execution_count":null,"metadata":{"_cell_guid":"b9bc9c0c-0fa1-da59-73cf-ed39517994ef"},"outputs":[],"source":"import pandas as pd\nimport numpy as np\nfrom shapely.wkt import loads\nfrom matplotlib.patches import Polygon\nimport matplotlib.pyplot as plt\n\ndf = pd.read_csv('../input/train_wkt.csv')\ndf.head()"},{"cell_type":"code","execution_count":null,"metadata":{"_cell_guid":"c233d5d6-68a0-2ae8-5757-d58291245163"},"outputs":[],"source":"# Use just first image\npolygonsList = {}\nimage = df[df.ImageId == '6100_1_3']\nfor cType in image.ClassType.unique():\n    polygonsList[cType] = loads(image[image.ClassType == cType].MultipolygonWKT.values[0])"},{"cell_type":"code","execution_count":null,"metadata":{"_cell_guid":"a3a22597-6823-e1f6-1902-369edf5a1baa"},"outputs":[],"source":"type(image)"},{"cell_type":"code","execution_count":null,"metadata":{"_cell_guid":"1fcab23c-506c-736d-86a4-32644a7226ce"},"outputs":[],"source":"# plot using matplotlib\nfig, ax = plt.subplots(figsize=(8, 8))\n\n# colors are going to be grayscale for now\nfor p in [8]:\n    for polygon in polygonsList[p]:\n        mpl_poly = Polygon(np.array(polygon.exterior), color=str(p/10.), lw=0, alpha=0.4)\n        ax.add_patch(mpl_poly)\n    \nax.relim()\nax.autoscale()"},{"cell_type":"code","execution_count":null,"metadata":{"_cell_guid":"1e343c4f-023a-ff75-0345-af4c7e280a3f"},"outputs":[],"source":""}],"metadata":{"_change_revision":0,"_is_fork":false,"kernelspec":{"display_name":"Python 3","language":"python","name":"python3"},"language_info":{"codemirror_mode":{"name":"ipython","version":3},"file_extension":".py","mimetype":"text/x-python","name":"python","nbconvert_exporter":"python","pygments_lexer":"ipython3","version":"3.5.2"}},"nbformat":4,"nbformat_minor":0}