Skip to content

Instantly share code, notes, and snippets.

@bennytowns
Created August 7, 2016 18:48
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save bennytowns/5393fca81519b0e296da75d23ce943d3 to your computer and use it in GitHub Desktop.
Save bennytowns/5393fca81519b0e296da75d23ce943d3 to your computer and use it in GitHub Desktop.
render_data <- function(DT,cat,Cond,texas,Cond1) {
if (Cond1==1900) {
Cond1=as.character(as.numeric(Cond)+1)
}
return_data = NULL
data_cat=NULL
data_cat_prev=NULL
for (i in 1:nrow(DT)) {
return_data$full[i]=levels(DT$full)[DT$full[i]]
return_data$state[i]=levels(DT$state)[DT$state[i]]
if (cat=="All Sources"){
data_cat = DT$NG_Gross_Withdrawals_All[[i]][Cond]
data_cat_prev = DT$NG_Gross_Withdrawals_All[[i]][Cond1]
}
if (cat=="Shale"){
data_cat = DT$NG_Gross_Withdrawals_From_Shale[[i]][Cond]
data_cat_prev = DT$NG_Gross_Withdrawals_From_Shale[[i]][Cond1]
}
if (cat=="Coal Beds"){
data_cat = DT$NG_Gross_Withdrawals_From_Coal_Beds[[i]][Cond]
data_cat_prev = DT$NG_Gross_Withdrawals_From_Coal_Beds[[i]][Cond1]
}
if (cat=="Oil Wells"){
data_cat = DT$NG_Gross_Withdrawals_From_Oil_Wells[[i]][Cond]
data_cat_prev = DT$NG_Gross_Withdrawals_From_Oil_Wells[[i]][Cond1]
}
if (cat=="Gas Wells"){
data_cat = DT$NG_Gross_Withdrawals_From_Gas_Wells[[i]][Cond]
data_cat_prev = DT$NG_Gross_Withdrawals_From_Gas_Wells[[i]][Cond1]
}
if (cat=="Residential"){
data_cat = DT$residential[[i]][Cond]
data_cat_prev = DT$residential[[i]][Cond1]
}
if (cat=="Industrial"){
data_cat = DT$industrial[[i]][Cond]
data_cat_prev = DT$industrial[[i]][Cond1]
}
if (cat=="Commercial"){
data_cat = DT$commercial[[i]][Cond]
data_cat_prev = DT$commercial[[i]][Cond1]
}
if (cat=="Electricity"){
data_cat = DT$electricity[[i]][Cond]
data_cat_prev = DT$electricity[[i]][Cond1]
}
if (cat=="Vehicles"){
data_cat = DT$vehicle[[i]][Cond]
data_cat_prev = DT$vehicle[[i]][Cond1]
}
if (cat=="Total"){
data_cat = DT$consumers[[i]][Cond]
data_cat_prev = DT$consumers[[i]][Cond1]
}
if (cat=="Consumption"){
data_cat = DT$Total_Consumption[[i]][Cond]
data_cat_prev = DT$Total_Consumption[[i]][Cond1]
}
return_data$sum[i]=sum(data_cat)
return_data$diff[i]=sum(data_cat_prev)-sum(data_cat)
}
return_data$sum[which(return_data$sum==0)]=NA
return_data$diff[which(return_data$diff==0)]=NA
if (texas==TRUE) {
return_data$sum[return_data$state=='TX']=NA
return_data$diff[return_data$state=='TX']=NA
}
return(return_data)
}
sortdata <- function(D,retnum) {
D = D[with(D, order(D$sum,decreasing=TRUE)), ]
return(D[1:retnum,])
}
sortdiffdata <- function(D,retnum) {
D = D[with(D, order(D$diff,decreasing=TRUE)), ]
D1 = D[with(D, order(D$diff,decreasing=FALSE)), ]
D = D[1:retnum,]
D1 = D1[retnum:1,]
return(rbind(D,D1))
}
chart_data <- function(DT,cat,cond,state) {
data_cat=NULL
for (i in 1:nrow(DT)) {
if (state==DT$state[i]) {
if (cat=="All Sources"){
data_cat = DT$NG_Gross_Withdrawals_All[[i]][cond]
}
if (cat=="Shale"){
data_cat = DT$NG_Gross_Withdrawals_From_Shale[[i]][cond]
}
if (cat=="Coal Beds"){
data_cat = DT$NG_Gross_Withdrawals_From_Coal_Beds[[i]][cond]
}
if (cat=="Oil Wells"){
data_cat = DT$NG_Gross_Withdrawals_From_Oil_Wells[[i]][cond]
}
if (cat=="Gas Wells"){
data_cat = DT$NG_Gross_Withdrawals_From_Gas_Wells[[i]][cond]
}
if (cat=="Total"){
data_cat = DT$consumers[[i]][cond]
}
if (cat=="Commercial"){
data_cat = DT$commercial[[i]][cond]
}
if (cat=="Residential"){
data_cat = DT$residential[[i]][cond]
}
if (cat=="Industrial"){
data_cat = DT$industrial[[i]][cond]
}
if (cat=="Electricity"){
data_cat = DT$electricity[[i]][cond]
}
if (cat=="Vehicles"){
data_cat = DT$vehicle[[i]][cond]
}
if (cat=="Consumption"){
data_cat = DT$Total_Consumption[[i]][cond]
}
if (cat=="Price"){
data_cat = DT$price[[i]][cond]
}
return(data_cat)
}
}
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment