Skip to content

Instantly share code, notes, and snippets.

/hidden_relationships.py

Created May 28, 2017 18:02
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save anonymous/d27e448edbea145dbe4db3362bbf75f2 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save anonymous/d27e448edbea145dbe4db3362bbf75f2 to your computer and use it in GitHub Desktop.
Download ZIP

Revisions

  1. @invalid-email-address Anonymous created this gist May 28, 2017.
    27 changes: 27 additions & 0 deletions hidden_relationships.py
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,27 @@
    import numpy as np
    import pandas as pd
    from sklearn.decomposition import PCA
    from sklearn.cluster import KMeans
    import matplotlib.pyplot as plt
    import seaborn as sb

    np.seterr(divide='ignore', invalid='ignore')

    # Quick way to test just a few column features
    # stocks = pd.read_csv('supercolumns-elements-nasdaq-nyse-otcbb-general-UPDATE-2017-03-01.csv', usecols=range(1,16))

    stocks = pd.read_csv('supercolumns-elements-nasdaq-nyse-otcbb-general-UPDATE-2017-03-01.csv')

    print(stocks.head())

    str_list = []
    for colname, colvalue in stocks.iteritems():
    if type(colvalue[1]) == str:
    str_list.append(colname)

    # Get to the numeric columns by inversion
    num_list = stocks.columns.difference(str_list)

    stocks_num = stocks[num_list]

    print(stocks_num.head())