Skip to content
Snippets Groups Projects
rank_order.py 1.92 KiB
Newer Older
import numpy as np
import pandas as pd
import csv
import matplotlib.pyplot as plt




tfsee = pd.read_csv('clustering_tfs.csv')

tfsee_cluster4 = tfsee[tfsee['cluster'] == 3]


tfsee_cluster4['early'] = tfsee_cluster4[['ES_D0','ES_D2','ES_D5']].mean(axis=1)
tfsee_cluster4['late'] = tfsee_cluster4[['ES_D7','ES_D10']].mean(axis=1)
tfsee_cluster4['diff'] = tfsee_cluster4['late'] - tfsee_cluster4['early']
tfsee_cluster4['rank'] = tfsee_cluster4['diff'].rank()

x = list(tfsee_cluster4['rank'])

z = np.polyfit(tfsee_cluster4['rank'], tfsee_cluster4['diff'], 3)
f = np.poly1d(z)
x_new = np.linspace(1, 36, num=len(x)*10)


plt.figure(figsize=(25,20))
plt.plot(x_new, f(x_new), color = 'k', linewidth=4.0)
plt.scatter(x=tfsee_cluster4['rank'], y=tfsee_cluster4['diff'], color='#3999CA', s=600)
plt.ylim([-0.5,3.5])
plt.suptitle('Late Differentiation Enriched TFS', fontsize=8, fontweight='bold')
plt.xlabel('Rank Order',fontsize=8, fontweight='bold')
plt.ylabel('delta z',fontsize=8, fontweight='bold')
plt.savefig('cluster4_enriched_tfs.png')
plt.clf()



tfsee_cluster3 = tfsee[tfsee['cluster'] == 2]


tfsee_cluster3['early'] = tfsee_cluster3[['ES_D0','ES_D2','ES_D5']].mean(axis=1)
tfsee_cluster3['late'] = tfsee_cluster3[['ES_D7','ES_D10']].mean(axis=1)
tfsee_cluster3['diff'] = tfsee_cluster3['early'] - tfsee_cluster3['late']
tfsee_cluster3['rank'] = tfsee_cluster3['diff'].rank()

x = list(tfsee_cluster3['rank'])

z = np.polyfit(tfsee_cluster3['rank'], tfsee_cluster3['diff'], 3)
f = np.poly1d(z)
x_new = np.linspace(1, 36, num=len(x)*10)


plt.figure(figsize=(25,20))
plt.plot(x_new, f(x_new), color = 'k', linewidth=4.0)
plt.scatter(x=tfsee_cluster3['rank'], y=tfsee_cluster3['diff'], color='#316A45', s=600)
plt.ylim([-0.5,3.5])
plt.suptitle('Early Enriched TFS', fontsize=8, fontweight='bold')
plt.xlabel('Rank Order',fontsize=8, fontweight='bold')
plt.ylabel('delta z',fontsize=8, fontweight='bold')
plt.savefig('cluster3_enriched_tfs.png')
plt.clf()