2a:T5d1,function make_chart(objData,objPredictedData) { const line = vl.markLine() .encode( vl.x().fieldT("date") .axis({ tickCount: 'month', format: '%B', grid: false, tickSize: 6 }) .title(null), // date parsing! vl.y().fieldQ("mean") .axis({ tickCount: 4, domain: false, ticks: false, labelAlign: 'left', labelBaseline: 'bottom' }) .title(dd1), vl.color({ value: 'blue' }) ) const predictedLine = vl.markLine().data(objPredictedData).transform( vl.window({ op: 'mean', field: dd1, as: 'mean' }).frame([-3, 3]), ) .encode( vl.x().fieldT("date") .axis({ tickCount: 'month', format: '%B', grid: false, tickSize: 6 }) .title(null), // date parsing! vl.y().fieldQ("mean") .axis({ tickCount: 4, domain: false, ticks: false, labelAlign: 'left', labelBaseline: 'bottom' }) .title(dd1), vl.color({ value: 'green' }) ) const bars = vl.markBar({ opacity: 0.1 }) .encode( vl.x().fieldT("date"), vl.y().fieldQ(dd1), vl.color({ value: 'red' }) ) return vl.layer(bars, line, predictedLine) .data(objData).transform( vl.window({ op: 'mean', field: dd1, as: 'mean' }).frame([-3, 3]), ).width(width).height(width / 5).autosize({ type: 'fit-x', contains: 'padding' }) .config({ style: { cell: { stroke: 'transparent' } }, axis: { labelColor: '#929292', gridColor: '#ececec' }, padding: { top: 5, right: 0, bottom: 5, left: 0 }, }).render(); }2b:[{"id":"50848f0dd4d2d9c84821adba90ca625cb2815694117bbb7b6ae393e788aae9d85f19a6c73d60d642bb04e2f95b87993e8c7312d4d4da6a1431c58460f5ad6c63","href":"https://static.observableusercontent.com/files/50848f0dd4d2d9c84821adba90ca625cb2815694117bbb7b6ae393e788aae9d85f19a6c73d60d642bb04e2f95b87993e8c7312d4d4da6a1431c58460f5ad6c63","name":"counties-albers-10m@1.json","type":"application/json","lastModified":1608926667147.335,"size":795407}] 29:{"title":"COVID-19 in the US","cells":[{"id":0,"value":"md`# COVID-19 in the US`","pinned":false,"hidden":false,"mode":"ojs"},{"id":619,"value":"viewof d = date({\n title: \"Choose date:\", \n min: uniqueDateArray[0],\n max: uniqueDateArray[uniqueDateArray.length-1],\n value: uniqueDateArray[uniqueDateArray.length-1],\n description: \"Only dates that COVID-19 data is available are allowed.\"\n})","pinned":false,"hidden":false,"mode":"ojs"},{"id":429,"value":"viewof dd1 = select({\n title: \"Variable to Visualize:\",\n description: \"Please pick your variable.\",\n options: [\"Cases\", \"Deaths\", \"New Cases\", \"New Deaths\"],\n value: \"Cases\"\n})","pinned":false,"hidden":false,"mode":"ojs"},{"id":537,"value":"viewof dd2 = select({\n title: \"Select State:\",\n description: \"Select State\",\n options: ['All', 'Alabama','Alaska','Arizona','Arkansas','California','Colorado','Connecticut','Delaware','District of Columbia','Florida','Georgia','Hawaii','Idaho','Illinois','Indiana','Iowa','Kansas','Kentucky','Louisiana','Maine','Maryland','Massachusetts','Michigan','Minnesota','Mississippi','Missouri','Montana','Nebraska','Nevada','New Hampshire','New Jersey','New Mexico','New York','North Carolina','North Dakota','Ohio','Oklahoma','Oregon','Pennsylvania','Rhode Island','South Carolina','South Dakota','Tennessee','Texas','Utah','Vermont','Virgin Island','Virginia','Washington','West Virginia','Wisconsin','Wyoming'\n],\n multiple: false\n})","pinned":false,"hidden":false,"mode":"ojs"},{"id":14,"value":"chart = {\n const svg = d3.create(\"svg\")\n .attr(\"viewBox\", [0, 0, 975, 610]);\n\n svg.append(\"g\")\n .attr(\"transform\", \"translate(610,20)\")\n .append(() => legend({color, title: data.title, width: 250}));\n\n svg.append(\"g\")\n .selectAll(\"path\")\n .data(topojson.feature(us, us.objects.counties).features)\n .join(\"path\")\n .attr(\"fill\", d => color(data.get(d.id)))\n .attr(\"d\", path)\n .append(\"title\")\n .text(d => `${d.properties.name}, ${states.get(d.id.slice(0, 2)).name}\n${format(data.get(d.id))}`);\n\n svg.append(\"path\")\n .datum(topojson.mesh(us, us.objects.states, (a, b) => a !== b))\n .attr(\"fill\", \"none\")\n .attr(\"stroke\", \"white\")\n .attr(\"stroke-linejoin\", \"round\")\n .attr(\"d\", path);\n\n return svg.node();\n}","pinned":false,"hidden":false,"mode":"ojs"},{"id":767,"value":"make_figure(dd2, objData, objPredictedData)","pinned":false,"hidden":false,"mode":"ojs"},{"id":762,"value":"make_figure = async (title, data, predictedData) => html`

${title}

\n ${await make_chart(data, predictedData)}\n`;","pinned":false,"hidden":false,"mode":"ojs"},{"id":753,"value":"$2a","pinned":false,"hidden":false,"mode":"ojs"},{"id":16,"value":"data = Object.assign(new Map(filteredStateData), {title: \"COVID-19 \"+dd1+\" in the US, \"+date_var})","pinned":true,"hidden":false,"mode":"ojs"},{"id":291,"value":"datareturn = path => {\nif (dd1 == \"Cases\" || dd1 == \"New Cases\") {\nvar df = d3.csv(path, ({fips, cases, date, state}) => [fips, +cases, date, state]);\n}\nelse if (dd1 == \"Deaths\" || dd1 == \"New Deaths\") {\nvar df = d3.csv(path, ({fips, deaths, date, state}) => [fips, +deaths, date, state]);\n}\nreturn df;\n}","pinned":true,"hidden":false,"mode":"ojs"},{"id":504,"value":"filteredDateDataReturn = x => {\nvar df = [];\nfor (var i = 0; i < x.length; i++) {\n if (x[i][2] == date_var) {\n df.push(x[i]);\n }\n}\n\nif (dd1 == \"New Cases\" || dd1 == \"New Deaths\") {\n \n var parts = date_var.split('-');\n \n var previous_day = String(parts[2] - 1);\n \n var date_var_previous = parts[0]+\"-\"+parts[1]+\"-\"+previous_day;\n \n var df_previous = [];\n \n for (var i = 0; i < x.length; i++) {\n if (x[i][2] == date_var_previous) {\n df_previous.push(x[i]);\n }\n}\n\nvar df_diff = [];\n \nfor (var i = 0; i < df.length; i++) {\n \n for (var j = 0; j < df_previous.length; j++) {\n \n if (df[i][0] == df_previous[j][0]) { df_diff.push([df[i][0],df[i][1]-df_previous[j][1],df[i][2],df[i][3]]);\n break; }\n \n }\n \n}\n \n}\n\nif (dd1 != \"New Cases\" && dd1 != \"New Deaths\") {\nreturn df;\n}\nelse {\nreturn df_diff;\n}\n}","pinned":true,"hidden":false,"mode":"ojs"},{"id":802,"value":"parts = date_var.split('-');","pinned":true,"hidden":false,"mode":"ojs"},{"id":804,"value":"previous_day = String(parts[2] -1);","pinned":true,"hidden":false,"mode":"ojs"},{"id":806,"value":"date_var_previous = parts[0]+\"-\"+parts[1]+\"-\"+previous_day;","pinned":true,"hidden":false,"mode":"ojs"},{"id":541,"value":"filteredStateDataReturn = x => {var df = []; if (dd2 != 'All') {for (var i = 0; i < x.length; i++) {\n if (x[i][3] == dd2) {\n df.push(x[i]);\n }\n}} else {df = x;}\nreturn df;}","pinned":true,"hidden":false,"mode":"ojs"},{"id":516,"value":"transpose = m => m[0].map((x,i) => m.map(x => x[i]))","pinned":true,"hidden":false,"mode":"ojs"},{"id":558,"value":"dataDF = datareturn(\"https://raw.githubusercontent.com/nytimes/covid-19-data/master/us-counties.csv\")","pinned":true,"hidden":false,"mode":"ojs"},{"id":591,"value":"transposedDF = transpose(dataDF)","pinned":true,"hidden":false,"mode":"ojs"},{"id":593,"value":"dateArray = transposedDF[2]","pinned":true,"hidden":false,"mode":"ojs"},{"id":599,"value":"uniqueDateArray = [...new Set(dateArray)]","pinned":true,"hidden":false,"mode":"ojs"},{"id":567,"value":"filteredDateData = filteredDateDataReturn(dataDF)","pinned":true,"hidden":false,"mode":"ojs"},{"id":571,"value":"filteredStateData = filteredStateDataReturn(filteredDateData)","pinned":true,"hidden":false,"mode":"ojs"},{"id":575,"value":"transposedData = transpose(filteredStateData)","pinned":true,"hidden":false,"mode":"ojs"},{"id":515,"value":"minmaxList = x => {var min = Math.min(...x); var max = Math.max(...x); if (dd2 == 'All') {return [min,max/34.0];} else {return [min,max/5.0];}}","pinned":true,"hidden":false,"mode":"ojs"},{"id":581,"value":"dataArray = transposedData[1]","pinned":true,"hidden":false,"mode":"ojs"},{"id":578,"value":"minmaxValues = minmaxList(dataArray)","pinned":true,"hidden":false,"mode":"ojs"},{"id":355,"value":"date_var = d","pinned":true,"hidden":false,"mode":"ojs"},{"id":69,"value":"color = {return d3.scaleSequential(minmaxValues, d3.interpolateReds);}","pinned":true,"hidden":false,"mode":"ojs"},{"id":199,"value":"path = d3.geoPath()","pinned":true,"hidden":false,"mode":"ojs"},{"id":75,"value":"format = d => `${d}`","pinned":true,"hidden":false,"mode":"ojs"},{"id":157,"value":"states = new Map(us.objects.states.geometries.map(d => [d.id, d.properties]))","pinned":true,"hidden":false,"mode":"ojs"},{"id":8,"value":"us = FileAttachment(\"counties-albers-10m@1.json\").json()","pinned":true,"hidden":false,"mode":"ojs"},{"id":6,"value":"topojson = require(\"topojson-client@3\")","pinned":true,"hidden":false,"mode":"ojs"},{"id":3,"value":"d3 = require(\"d3@5\", \"d3-regression@1\")","pinned":true,"hidden":false,"mode":"ojs"},{"id":176,"value":"import {legend} from \"@d3/color-legend\"","pinned":true,"hidden":false,"mode":"ojs"},{"id":392,"value":"import {select} from \"@jashkenas/inputs\"","pinned":true,"hidden":false,"mode":"ojs"},{"id":451,"value":"import {date} from \"@jashkenas/inputs\"","pinned":true,"hidden":false,"mode":"ojs"},{"id":606,"value":"import { slider as slide } from '@lorenries/inputs'","pinned":true,"hidden":false,"mode":"ojs"},{"id":629,"value":"import {vl} from '@vega/vega-lite-api'","pinned":true,"hidden":false,"mode":"ojs"},{"id":878,"value":"import { slider } from \"@jashkenas/inputs\";","pinned":true,"hidden":false,"mode":"ojs"},{"id":635,"value":"date_extent = [uniqueDateArray[0],uniqueDateArray[uniqueDateArray.length-1]]","pinned":true,"hidden":false,"mode":"ojs"},{"id":646,"value":"stateData = filteredStateDataReturn(dataDF)","pinned":true,"hidden":false,"mode":"ojs"},{"id":650,"value":"sumdata = x => {var sumDF = []; var ids = []; for (var i = 0; i < uniqueDateArray.length; i++) { var sum = 0.0; for (var j = 0; j < x.length; j++) { if (x[j][2] == uniqueDateArray[i]) { sum = sum + x[j][1]; } } sumDF.push(sum); ids.push(i); } return [sumDF,ids];}","pinned":true,"hidden":false,"mode":"ojs"},{"id":830,"value":"ids = sumdata(stateData)[1]","pinned":true,"hidden":false,"mode":"ojs"},{"id":656,"value":"sumStateData = {if (dd1 != \"New Cases\" && dd1 != \"New Deaths\") {return sumdata(stateData)[0];} else { var smd = sumdata(stateData)[0]; var diffData = []; for (var i = 1; i < smd.length; i++) { diffData.push(smd[i] - smd[i-1]); } return diffData; }}","pinned":true,"hidden":false,"mode":"ojs"},{"id":660,"value":"combinedData = transpose([uniqueDateArray, sumStateData, ids])","pinned":true,"hidden":false,"mode":"ojs"},{"id":669,"value":"convertToObject = x => {var objs = []; for (var i = 0; i < x.length; i++) { var obj = {}; obj['date'] = x[i][0]; obj[dd1] = x[i][1]; obj['id'] = x[i][2]; objs.push(obj);} return objs;}","pinned":true,"hidden":false,"mode":"ojs"},{"id":748,"value":"objData = convertToObject(combinedData)","pinned":true,"hidden":false,"mode":"ojs"},{"id":932,"value":"mlmodule = require(\"https://www.lactame.com/lib/ml/4.0.0/ml.min.js\")","pinned":true,"hidden":false,"mode":"ojs"},{"id":934,"value":"regression = new mlmodule.RandomForestRegression({\n seed: 3,\n maxFeatures: 1,\n replacement: false,\n nEstimators: 10000,\n useSampleBagging: true\n})","pinned":true,"hidden":false,"mode":"ojs"},{"id":957,"value":"convertTo2D = x => { var x2D = []; for (var i = 0; i < x.length; i++) { x2D.push([x[i]]); } return x2D; }","pinned":true,"hidden":false,"mode":"ojs"},{"id":961,"value":"ids2D = convertTo2D(ids)","pinned":true,"hidden":false,"mode":"ojs"},{"id":976,"value":"function slicer(x,start,end) { var slicedX = new Array(x.length); if (dd1 == \"New Cases\" || dd1 == \"New Deaths\") { slicedX = x.slice(start,end); } else { slicedX = x; } return slicedX; }","pinned":true,"hidden":false,"mode":"ojs"},{"id":1053,"value":"sIds2D = slicer(ids2D,1,ids2D.length)","pinned":true,"hidden":false,"mode":"ojs"},{"id":942,"value":"regression.train(sIds2D, sumStateData)","pinned":true,"hidden":false,"mode":"ojs"},{"id":945,"value":"result = regression.predict(sIds2D)","pinned":true,"hidden":false,"mode":"ojs"},{"id":965,"value":"combinedPredictedData = transpose([uniqueDateArray, result, ids])","pinned":true,"hidden":false,"mode":"ojs"},{"id":967,"value":"objPredictedData = convertToObject(combinedPredictedData)","pinned":true,"hidden":false,"mode":"ojs"}],"files":"$W2b","stdlib":"1","titling":"auto"}