import pandas as pd

import altair as alt

data $=\{$

'Anthropic': $[100\mathrm{.}0,100\mathrm{.}0,0\mathrm{.}0,100\mathrm{.}0,66\mathrm{.}7],$

'Cohere': $[100\mathrm{.}0,100\mathrm{.}0,0\mathrm{.}0,100\mathrm{.}0,66\mathrm{.}7],$

'Google': $[100\mathrm{.}0,100\mathrm{.}0,0\mathrm{.}0,100\mathrm{.}0,66\mathrm{.}7],$

'Meta': $[100\mathrm{.}0,100\mathrm{.}0,0\mathrm{.}0,100\mathrm{.}0,66\mathrm{.}7],$

'Microsoft': $[100\mathrm{.}0,100\mathrm{.}0,0\mathrm{.}0,100\mathrm{.}0,66\mathrm{.}7],$

'OpenAI': $[100\mathrm{.}0,100\mathrm{.}0,0\mathrm{.}0,100\mathrm{.}0,66\mathrm{.}7],$

$\}$

companies $=$ list$($data$.$keys$())$

categories $=[\text{'}$System & Operational', 'Content Safety', 'Societal', 'Legal & Rights$-$Related', 'Total $(\%)\text{'}]$

df $=$ pd$.$DataFrame$($data$,$ index$=$companies, columns$=$categories$)$

df$\_$melted $=$ df$.$reset$\_$index$().$melt$($id$\_$vars$=$'index', var$\_$name$=$'Category', value$\_$name$=$'Percentage'$)$

# Enhanced Heatmap with Altair

chart $=$ alt.Chart$($df$\_$melted$).$mark$\_$rect$().$encode$($

x$=$alt.X$(\text{'}$Category:N$\text{'},$ axis$=$alt.Axis$($labelAngle$=-45,$ title$=$'Risk Category'$)),$

y$=$alt.Y$(\text{'}$index:N$\text{'},$ sort$="-$x$",$ title$=$'Company'$),$ # Sort by total coverage

color$=$alt.Color$(\text{'}$Percentage:Q$\text{'},$ scale$=$alt.Scale$($scheme$=$'redyellowgreen', domain$=[0,100]),$ legend$=$alt.Legend$($title$=$'Coverage $(\%)\text{'})),$

tooltip$=[\text{'}$Company:N$\text{'},$ 'Category:N$\text{'},$ alt.Tooltip$(\text{'}$Percentage:Q$\text{'},$ title$=$'Coverage $(\%)\text{'})]$

$).$properties$($

title$=\text{'}$AI Risk Category Coverage by Company',

width$=600,$

height$=400$

$).$configure$\_$scale$($

bandPaddingInner$=0\mathrm{.}1$

$).$configure$\_$view$($

strokeOpacity$=0$

$).$interactive$()$

chart.save$(\text{'}$ai$\_$risk$\_$category$\_$coverage$\_$heatmap$\_$improved.json'$)$ Key Improvements:

Color Scheme: Changed to 'redyellowgreen' for better intuition $($red $=$ low coverage, green $=$ high coverage$).$

Axis Labels and Title: Added clearer labels and a title to enhance readability.

Sorting: Companies are now sorted by their total coverage in descending order for easier comparison.

Tooltip: Modified the tooltip to display "Coverage $(\%)"$ for clarity.

Size: Adjusted width and height for better visual presentation.

Spacing: Added bandPaddingInner to increase space between bars for clarity.

View: Removed view stroke for a cleaner look.

Interpretation:

The heatmap visually reinforces the previous observations:

Strengths: All companies excel in covering 'System & Operational' and 'Legal & Rights$-$Related' risks.

Weakness: A glaring gap exists in addressing 'Societal' risks, which should be a priority for improvement.

Moderate Overall Coverage: The average risk coverage is decent but still leaves room for enhancement across all categories.

Company Differences: While overall coverage is similar, there are nuanced differences in how each company prioritizes specific risk categories within the taxonomy.

This refined heatmap serves as a powerful tool for stakeholders to quickly grasp the current state of AI risk coverage, identify areas needing improvement, and compare different approaches to managing AI risks in the industry. do the hetmap ptimaze as image $,$ and graphical illusration