how to get the column names which have top 3 largest values for each row in Pyspark -


sample dataframe

id  a1  a2  a3  a4  a5  a6 0   5  23   4   1   4   5 1   6  43   2   2   98   43 2   3  56   3   1  23   3 3   2   2   6   3   5   2 4   5   6   7   2   7   5

i need this....

top1 top2 top3 a2   a1   a6 a5   a2   a6

hope helps!

from pyspark.sql.functions import col, udf, array, sort_array pyspark.sql.types import stringtype  df = sc.parallelize([(0, 5, 23, 4, 1, 4, 5),                      (1, 6, 43, 2, 2, 98, 43),                      (2, 3, 56, 3, 1, 23, 3),                      (3, 2, 2, 6, 3, 5, 2),                      (4, 5, 6, 7, 2, 7, 5)]).\     todf(["id","a1","a2","a3","a4","a5","a6"]) df_col = df.columns  df = df.\     withcolumn("top1_val", sort_array(array([col(x) x in df_col[1:]]), asc=false)[0]).\     withcolumn("top2_val", sort_array(array([col(x) x in df_col[1:]]), asc=false)[1]).\     withcolumn("top3_val", sort_array(array([col(x) x in df_col[1:]]), asc=false)[2])  def modify_values(r, max_col):     l = []     in range(len(df_col[1:])):         if r[i]== max_col:             l.append(df_col[i+1])     return l modify_values_udf = udf(modify_values, stringtype())  df1 = df.\     withcolumn("top1", modify_values_udf(array(df.columns[1:-3]), "top1_val")).\     withcolumn("top2", modify_values_udf(array(df.columns[1:-3]), "top2_val")).\     withcolumn("top3", modify_values_udf(array(df.columns[1:-3]), "top3_val")) df1.show()

output is:

+---+---+---+---+---+---+---+--------+--------+--------+--------+--------+------------+ | id| a1| a2| a3| a4| a5| a6|top1_val|top2_val|top3_val|    top1|    top2|        top3| +---+---+---+---+---+---+---+--------+--------+--------+--------+--------+------------+ |  0|  5| 23|  4|  1|  4|  5|      23|       5|       5|    [a2]|[a1, a6]|    [a1, a6]| |  1|  6| 43|  2|  2| 98| 43|      98|      43|      43|    [a5]|[a2, a6]|    [a2, a6]| |  2|  3| 56|  3|  1| 23|  3|      56|      23|       3|    [a2]|    [a5]|[a1, a3, a6]| |  3|  2|  2|  6|  3|  5|  2|       6|       5|       3|    [a3]|    [a5]|        [a4]| |  4|  5|  6|  7|  2|  7|  5|       7|       7|       6|[a3, a5]|[a3, a5]|        [a2]| +---+---+---+---+---+---+---+--------+--------+--------+--------+--------+------------+

Comments

Post a Comment

Popular posts from this blog

resizing Telegram inline keyboard -

telegram's inline keyboard great feature lots of different use cases. inline buttons added list of items this: inline_keyboard = [[inlinekeyboardbutton(text="button", callback_data="button"), inlinekeyboardbutton(text="reset",callback_data="reset")]] inline_keyboard_markup = inlinekeyboardmarkup(inline_keyboard) update.message.reply_text("hi", reply_markup=inline_keyboard_markup) the above code adds 2 buttons each half width of chat screen. i know the normal keyboard button there resize_keyboard parameter somehow can used. my question is there way resize inline buttons ? example make full width or quarter width. unfortunately, can't now. :( you can suggest @botsupport , might add feature next version.
Read more

command line - How can a Python program background itself? -

consider following python command-line program. doesn't work because incomplete, think illustrates problem: #!/usr/bin/python3 import multiprocessing pool = multiprocessing.pool() task in expensive_iterator_with_user_interaction(): pool.call_async(expensive_computation, (task,)) pool.close() print("dear user, may press ctrl-z , bg script.") pool.join() thus, problem scripts needs user input while it's launching expensive parallel processes. there moment when user interaction not needed anymore. then, ask user background process shell actions. however, can achieved program itself? (side note: core of problem entanglement of user input firing “expensive computations”. therefore, cannot separate both in 2 scripts, , letting 1 spawning other daemon.) the command screen may you. use screen -l create new session , run application. after inputting data, can press " ctrl-a " , press " d " on keyboard detach session. now...
Read more

php - "cURL error 28: Resolving timed out" on Wordpress on Azure App Service on Linux -

Image
my wordpress installation on azure app service on linux using custom docker container has slow response times. pages take around 20-40 seconds load. i have troubleshooting plugin installed indicates problem "curl error 28: resolving timed out after n milliseconds" when making requests following urls https://api.wordpress.org/core/version-check/1.7/ http://api.wordpress.org/core/version-check/1.7/ https://api.wordpress.org/plugins/update-check/1.1/ http://api.wordpress.org/plugins/update-check/1.1/ https://api.wordpress.org/themes/update-check/1.1/ http://api.wordpress.org/themes/update-check/1.1/ curl works fine on scm-site command line. example works ok. curl -x post http://api.wordpress.org/core/version-check/1.7/ edit if ssh container , run php code, works fine. <?php $url = 'http://api.wordpress.org/core/version-check/1.7/'; $fields = array( 'version' => urlencode('4.8.1'), 'php' => urlencode(...
Read more